Блог

Мой текущий код:

 import pandas as pd

df = pd.DataFrame.from_csv('data.csv',index_col=False)


while True:
     print (df)
     print ('Select X-Axis')
     xaxis = input()
     print ('Select Y-Axis')
     yaxis = input()
     break
 
df.plot(x= xaxis, y= yaxis)  

Мой текущий работает, только если пользователь вводит полное имя столбца. Однако некоторые столбцы большие. Я хочу упростить задачу для моего пользователя, чтобы, если он введет часть имени столбца, это сработало. Или, если каким-то образом я могу присвоить каждому имени столбца номер, не просматривая вручную каждый столбец и не присваивая номер. Затем я могу показать список пользователю, и он вводит число, которое он хотел бы отобразить.

Один из способов сделать это — использовать filter в pandas для этой цели.

 df.filter(regex=(yaxis))
  

Он отобразит все столбцы, соответствующие этой подстроке yaxis

вот пример.

 A = { 'Name': [ 'John', 'Andrew', 'Smith'] , 'Age' : [20,23,42]}

A
Out[19]: {'Age': [20, 23, 42], 'Name': ['John', 'Andrew', 'Smith']}

df = pd.DataFrame(A)

df
Out[21]: 
   Age    Name
0   20    John
1   23  Andrew
2   42   Smith

df.filter(regex=('Na'))
Out[22]: 
     Name
0    John
1  Andrew
2   Smith

df.filter(regex=('e'))
Out[23]: 
   Age    Name
0   20    John
1   23  Andrew
2   42   Smith