包含包含的熊猫数据框搜索并获取行号
在数据框中,我想在第一列中搜索一个字符串,并在找到该字符串时,我想获取行号并将其分配为变量,
我的数据框看起来像这样
df = pd.DataFrame({'A': ['xx', '--- bios ---', 'yy', 'zz', '--- os ---'],'B': ['rat', '', 'winter', 'host','']})
我正在尝试获取字符串的行号并将其分配给一个变量,如下所示
var = df.iloc[:,0].str.contains('--- bios ---').index
其中 --- bios --- 是搜索词,我正在尝试获取索引
但我没有得到所需的输出,我期望的输出是 4,这是行号
回答
df.iloc[:,0].str.contains('--- bios ---')仍然具有与 相同的(完整)索引df。你可能想要:
df.index[df.iloc[:,0].str.contains('--- bios ---')]
或者:
df[df.iloc[:,0].str.contains('--- bios ---')].index