如何从csv文件中检索特定的行和列

时间:2019-03-08 16:55:22

标签: python python-3.x pandas

我有两个具有用户名和不同ID的csv文件。根据用户的输入,我应该能够切换其ID并进行检索。例如。用户输入了学生ID,并希望我可以检索到特定人名的员工ID1。我对编程还是很陌生,所以对编程我不太了解。我该怎么做呢?请帮我 我的文件样本

  

姓名,Student_id,EmployeeId1,Employee_Id2

import pandas as pd
df = pd.read_csv("my1.csv")
colls = input("Enter the column")
roww = input("Enter the row")
df.loc ["roww","colls"] 

用户将不知道行和列,但我只是想尝试,但是没有用。

1 个答案:

答案 0 :(得分:1)

您正在寻找带有标签"roww"的行,而不是变量roww中输入的内容。

尝试删除引号,即:

df.loc [roww, colls] 

您可能还存在行可能是数字的问题,默认情况下,input中的值是字符串。在这种情况下,请尝试:

df.loc [int(roww),colls]

例如,如果您的csv文件为:

Name, Student_id,EmployeeId1, Employee_Id2
Bob, 1, 11, 12
Steve, 2, 21, 22

然后在列中输入Name,在行中输入1,将得到Steve作为答案(因为行从0开始。