python - 根据熊猫中的索引范围组合列的行

Question

我有一个包含一列的数据框。

Index | column1 |
0         and
1         too
2         ask
3         the
4         but
5         hat
6         hot
7         top
8         tap

我想根据条件组合索引之间的行。例如，如果一行有字母“a”，则索引将是：

0, 2, 5, 8

因此，组合行：

(0, 1), (2, 3, 4), (5, 6, 7), (8)

最后的输出是：

Index | column1 |
0         and, too
1         ask, the, but
2         hat, hot, top
3         tap

我尝试过的是：

[i for i in range(len(df['column1'])) if 'a' in df['column1'][i]]

给我指数：

[0, 2, 5, 8]

但从这里卡住了。谢谢

标签： pythonpandaslistlist-comprehension

Answer 1

比较 bya并Series.str.contains创建组 by ，然后通过过滤和最后一个聚合Series.cumsum删除可能的第一组包含非值：ag[g > 0]join

g = df['column1'].str.contains('a').cumsum()

df = df.groupby(g[g > 0])['column1'].apply(', '.join).reset_index(drop=True).to_frame()
print (df)
         column1
0       and, too
1  ask, the, but
2  hat, hot, top
3            tap

第一个值不包含a：

print (df)
  column1
1     too
2     ask
3     the
4     but
5     hat
6     hot
7     top
8     tap

g = df['column1'].str.contains('a').cumsum()

df = df.groupby(g[g > 0])['column1'].apply(', '.join).reset_index(drop=True).to_frame()
print (df)
         column1
0  ask, the, but
1  hat, hot, top
2            tap