在行pandas python上使用部分字符串匹配返回DataFrame项目[重复]
问题内容:
这个问题已经在这里有了答案 :
从pandas DataFrame中按部分字符串选择 (11个答案)
去年关闭。
我有两列的数据框,其中一列为字符串,另一列为列表,如下所示:
RSD_TYPE FILTER LIST
0 AQ500 [N/A, Z mean, SNR mean, Dir mean]
1 Triton [wipe mean, Z mean, Avail mean, Dir mean]
2 Windcube [N/A, W mean, Q mean, Dir mean]
3 Zephir [Rain mean, W mean, Packets, dir mean]
我想基于部分字符串与列RSD_TYPE的元素匹配返回一个列表。EG搜索哪行具有与“ AQ5”匹配的部分字符串,然后从该行返回相应的列表项,在这种情况下为[N
/ A,Z均值,SNR均值,Dir均值]。
计划是使用.get_value进行此操作,但首先我需要一种使用部分字符串匹配返回(行)索引的方法。那就是我被困住的地方。我知道如何在列标题上运行部分字符串匹配,但是我找不到在该列中的元素(或整个数据框)上运行它的方法。有任何想法吗?
提前谢谢了。
问题答案:
尝试这个:
df[df['RSD_TYPE'].str.contains("AQ5")]['FILTER LIST']
例:
In [3]: df
Out[3]:
RSD_TYPE FILTER LIST
0 AQ500 [N/A, Z mean, SNR mean, Dir mean]
1 Triton [wipe mean, Z mean, Avail mean, Dir mean]
2 Windcube [N/A, W mean, Q mean, Dir mean]
3 Zephir [Rain mean, W mean, Packets, dir mean]
[4 rows x 2 columns]
In [4]: df[df['RSD_TYPE'].str.contains("AQ5")]
Out[4]:
RSD_TYPE FILTER LIST
0 AQ500 [N/A, Z mean, SNR mean, Dir mean]
[1 rows x 2 columns]
In [5]: df[df['RSD_TYPE'].str.contains("AQ5")]['FILTER LIST']
Out[5]:
0 [N/A, Z mean, SNR mean, Dir mean]
Name: FILTER LIST, dtype: object