python dataframe 两列都是set,怎么做合并
时间: 2024-05-09 07:14:42 浏览: 126
python pandas dataframe 按列或者按行合并的方法
5星 · 资源好评率100%
可以使用 Pandas 库中的 apply 函数,将两列的 set 合并成一个新的 set,然后赋给一个新的列。示例代码如下:
```python
import pandas as pd
# 创建示例 dataframe
df = pd.DataFrame({'A': [{'a', 'b'}, {'c', 'd'}],
'B': [{'b', 'c'}, {'d', 'e'}]})
# 定义合并函数
def merge_sets(row):
return row['A'].union(row['B'])
# 使用 apply 函数将合并后的 set 赋给新的列 C
df['C'] = df.apply(merge_sets, axis=1)
print(df)
```
运行结果如下:
```
A B C
0 {a, b} {b, c} {a, b, c}
1 {c, d} {d, e} {c, d, e}
```
其中,apply 函数的参数 axis=1 表示按行执行合并函数,即对每一行的 A 和 B 列执行 merge_sets 函数。函数 merge_sets 中使用了 set 的 union 方法将两个 set 合并成一个新的 set,并返回给 apply 函数。最终,apply 函数将每一行的合并结果赋给新的列 C。
阅读全文