熊猫-获取行匹配条件的列值

时间:2020-05-22 22:42:47

标签: python pandas

我正在Pandas中读取CSV文件。假设CSV如下:

column_1,column_2,column_3
1,2,value_1
1,3,value_2
2,1,value_3
2,2,value_4

我想从column_3value_2的{​​{1}}(即column_1=1)中获取值。我确定只有1行符合此条件。

所以我正在做这样的事情:

column_2=3

我尝试了df = pd.read_csv(...file...) cond = (df['column_1'] == 1) & (df['column_2'] == 3) ... 给我值,但是它返回一个数据帧,其中索引为该行的行号。这里的行号不是0,而是1(即CSV文件中的原始行号),这不允许我执行df.loc[cond, 'column_3']

如何在此处获取列值?

1 个答案:

答案 0 :(得分:0)

这应该有效:

import pandas as pd

data = {
    'column_1': [1, 1, 2, 2],
    'column_2': [2, 3, 1, 2],
    'column_3': ['value_1', 'value_2', 'value_3', 'value_4']
}

df = pd.DataFrame(data=data)

cond = (df['column_1'] == 1) & (df['column_2'] == 3)
result = df[cond].column_3.values[0]
result