Question

假设我有一个打开的csv文件，如下所示：

代码：

import pandas as pd
table = pd.read_csv("numbers.csv")
user_input = input()
print(table)

输出：

    year  number
0   2001   1
1   2002   41
2   2003   123
3   2004   135

基本上，我试图这样做，以便当用户输入的年份与年份匹配时，它将返回同一行中的数字。

例如：用户输入= 2002 输出= 41

不知道如何进行任何帮助！

Answer 1

您可以尝试：

import pandas as pd
table = pd.read_csv("numbers.csv")
user_input = int(input())
#print(table)

table.set_index('year', inplace=True)
print(table.loc[user_input]['number'])

如何从下一列中选择一个熊猫值

1 个答案: