python - 从不同的列中按键获取值

Question

我有一个包含“键”和“值”列的数据框，我想输入一个键并输出一个值。

import pandas as pd

df = pd.DataFrame({
    'key': ['A', 'B', 'C', 'D'],
    'value': [2, 8, 10, 12]})

print(df)

key = 'B'
value = df[df['key'] == key]['value']
print(value)

当前输出如下：

  key  value
0   A      2
1   B      8
2   C     10
3   D     12
1    8
Name: value, dtype: int64

如何获得输出值：在这种情况下为 8，因为键是“B”。

Answer 1

您必须使用 df.loc 来获取特定值。你可以给这样的东西。

df.loc[df.key == 'B','value'].item()

这个输出将是：

8

或者你也可以给：

df.loc[df.key == 'B']['value'].item()

或者你可以给：

df[df.key == 'B']['value'].iloc[0]

或者你可以给：

df.loc[df.key == 'B', 'value'].values[0]

如果您确定数据框中只有一个带有键 'B' 的项目，请使用 .item()。如果您不确定，那么我建议您使用.values[0]，因为它会从结果中选择第一个。

所有这些都会给你一个8的值

Answer 2

试试这个。这很简单。

print(df['B'])

Answer 3

这应该够好了！！

df[df['key']=='B']['value'].iloc[0]