dataframe groupby 取字符串列第一个
时间: 2023-10-13 19:10:03 浏览: 116
My SQL group by取同组第一条
可以使用 `groupby` 和 `apply` 方法来实现。假设需要对 DataFrame 中的列 `col1` 进行分组,取出每组中 `col2` 列的第一个字符串,可以按如下方式编写代码:
``` python
df.groupby('col1')['col2'].apply(lambda x: x.str[0])
```
上述代码中,`lambda` 函数 `x.str[0]` 取出了 `col2` 中每个字符串的第一个字符。`apply` 方法应用这个函数到每个分组中的 `col2` 列,最终输出一个 Series,其中每个元素都是字符串的第一个字符。
阅读全文