python - 如果其他列的值与子字符串匹配而不仅仅是相等比较，如何提取 pandas df 单元格值

Question

这个问题已经有答案了:

Extract column value based on another column in Pandas (7 个回答)

已关闭 4 年前。

所以我需要提取另一列与模式匹配的行的单元格值。

我有以下数据框 df:

id    A               B           C
0  2018-A4-X     some_name  2018-09-02
1  2018-A3-X     some_name  2018-05-13
2  2018-A2-X     some_name  2018-02-18
3  2018-A1-X     some_name  2017-11-26

我想提取 C 列的值，其中 A 列的值包含字符串“A4”。

我尝试这样做，但它给出了“KeyError:True”。

a4 = df.loc['A4' in str(df['id']), 'C'].iloc[0]

所以我在上述情况下的输出应该是 2018-09-02

Answer 1

用途:

print(df.loc[df['A'].str.contains('A4'),'C'].item())

输出:

2018-09-02

使用 loc，然后使用 str.contains 到 'A' 列来查看 'A4' 是否为如果是，则获取'C'列，并使用item获取值。