Если оператор использует столбец pandas в качестве условия и пользовательский ввод - PullRequest
Новая
       13

Если оператор использует столбец pandas в качестве условия и пользовательский ввод

1 голос
/ 16 марта 2020

У меня много проблем, чтобы решить эту проблему: я хочу использовать пользовательский ввод для поиска в наборе данных и дать пользователям полную строку, совпадающую с ее вводом. Код, который я сделал, выглядит примерно так, но я не знаю, как исправить проблемы в операторе if, любое предложение будет отличным. 

import pandas as pd

a = pd.read_csv('data.txt', delimiter = '\t')

pd.set_option('display.max_rows', None)
pd.set_option('display.max_columns', None)
pd.set_option('display.max_colwidth', None)

a.set_index('Rank', inplace = True)
user_rank = input('Choose the rank: ')

if int(user_rank) == all(a.iloc['Rank']):
  print(a.loc[user_rank])
else:
  print('Error')
...