如何根据 pandas Dataframe 的条件打印值？

Question

我是 pandas dataframe 和 python 的新手。 目前，我有一个 pandas dataframe ，我想根据设置的条件打印值。

我的 df 看起来像这样：

ID     Name    Price
1      Apple     1
2      Orange    0.5
3      Pear      0.7

我想对其进行编码，以便当我要求用户输入 ID 时，它会返回价格。 例如，如果用户输入 2，它应该返回 0.5。

inputid = input("Please input ID: ")

接下来我应该怎么做才能根据输入从 df 获取返回价格？

Answer 1

可能的解决方案之一：

1. 将df中的ID列设置为其索引：

    df.set_index('ID', inplace=True)

2. 定义如下 function：

    def getPrice(): inputid = input("Please input ID: ") try: return df.loc[int(inputid), 'Price'] except: return 'No such ID'

然后当你调用它时，执行getPrice() ：

• 将显示输入提示。
• 用户输入 ID。
• 在try块中，此 function 尝试：
  • 将inputid转换为int （索引包含int值，但inputid是string ），因此导致错误的一个原因可能是用户只是按Enter而不输入任何值，
  • 即使用户输入了一个数字，也有可能df不包含这样的ID （错误的第二个原因）。
• 如果到目前为止一切正常，function 将返回感兴趣的价格。
• 但是，如果发生任何错误，则结果是错误消息。

Answer 2

Pandas 文档非常好，将涵盖您需要的一切。

https://pandas.pydata.org/pandas-docs/stable/user_guide/indexing.html

让我们调用您的 DataFrame price_df ，具体取决于您的 ID 列是否设置为 dataframe 的索引。

如果是：

print(price_df.loc[inputid, "Price"])

如果不是：

print(price_df[price_df["ID"] == inputid]["Price"])

.loc方法是从 DataFrame 访问数据的主要方式。

Answer 3

这是一个可行的解决方案。

df = pd.DataFrame({'ID' : (1, 2,3),
                  'Name':('Apple', 'Orange', 'Pear'),
                  'Price': (1, 0.5, 0.7)})
df
    ID  Name    Price
0   1   Apple   1.0
1   2   Orange  0.5
2   3   Pear    0.7

userin = input("Please input ID: ")
userin
`1`

df.loc[df['ID'] == float(userin)]

    ID  Name    Price
0   1   Apple   1.0

您需要将用户输入更改为float或int以便 pandas 读取。

Answer 4

你可以试试：

userin = input("input ID: ")

print("The price for the ID is- ",df[df.ID==float(userin)].Price.values[0])