如何使用 python 在数据框中选择包含某些特定值的行？

Question

我对 python 和编码很陌生，如果我可能不太清楚，请提前抱歉。 我有一个数据框，其中行对应于 ID（f.ied），列对应于多个值（ICD10 代码）。我想选择包含特定 ICD10 代码的行。 但是，我找不到正确的方法...我尝试了 loc 和 set 但没有运气...有什么帮助吗？

数据框是这样的：

每一行对应于 f.ied (ID)。我想知道哪个 f.ied 有特定的代码：I20、I21、I22、I23、I24、I25。

Answer 1

df = pd.DataFrame({'feid': [2, 4, 8, 0],
                    'f42002': [2, 0, 0, 0],
                    'f42003': [10, 'I21', 1, 'J10']})

df = df.set_index('feid')
df

数据帧

    f42002  f42003
feid        
2   2   10
4   0   I21
8   0   1
0   0   J10

想要的项目

mylist = ['I21', 'J10']

for i in mylist:
  print(df[(df['f42002']==i) | (df['f42003']==i)].index.values)

结果：

[4]
[0]