我想在包含給定字符串的數據幀（“類別”）中查找列的名稱。categories Groceries Electricity Fastfood Parking 0 SHOP ELCOMPANY MCDONALDS park1 MARKET ELECT Subway car2 market electr Restauran 247 假設我想在整個數據框中搜索字符串“MCDO”。答案應該是“快餐”。我嘗試使用 str.contains 但它似乎不適用于數據幀。我怎樣才能做到這一點？謝謝你。
3 回答
青春有我
TA貢獻1784條經驗 獲得超8個贊
如果您可以搜索整個字符串，那就更容易了，
(df == 'MCDONALDS').any().idxmax()
否則使用申請，
df.apply(lambda x: x.str.startswith('MCDO').any()).idxmax()
白衣非少年
TA貢獻1155條經驗 獲得超0個贊
也可以for為此使用循環：
def strfinder(df, mystr):
for col in df:
for item in df[col]:
if mystr in item:
return col
print(strfinder(df, 'MCDO'))
要獲取可能包含字符串的所有列，例如在下面修改后的數據框中：
Groceries Electricity Fastfood Parking
0 SHOP ELCOMPANY MCDONALDS park
1 MARKET MCDON Subway car
2 market electr Restauran 247
可以使用“列表理解”：
mystr = 'MCDO'
outlist = [ col
for col in df
for item in df[col]
if mystr in item ]
print(outlist)
輸出：
['Electricity', 'Fastfood']
翻翻過去那場雪
TA貢獻2065條經驗 獲得超14個贊
或使用：
print(df.apply(lambda x: x.str.contains('MCDO')).replace(False,np.nan).dropna(axis=1,how='all').columns.item())輸出：
Fastfood
添加回答
舉報
0/150
提交
取消