python中使用contact函数时可以设置列名嘛
时间: 2024-01-16 15:05:29 浏览: 66
Python中的匿名函数使用简介
在 Python 中使用 `pandas` 库的 `concat()` 函数时,可以使用 `keys` 参数来设置列名。具体实现方法如下:
```
import pandas as pd
# 创建两个 DataFrame
df1 = pd.DataFrame({'A': ['A0', 'A1', 'A2', 'A3'],
'B': ['B0', 'B1', 'B2', 'B3'],
'C': ['C0', 'C1', 'C2', 'C3'],
'D': ['D0', 'D1', 'D2', 'D3']})
df2 = pd.DataFrame({'A': ['A4', 'A5', 'A6', 'A7'],
'B': ['B4', 'B5', 'B6', 'B7'],
'C': ['C4', 'C5', 'C6', 'C7'],
'D': ['D4', 'D5', 'D6', 'D7']})
# 使用 concat() 函数合并两个 DataFrame,设置列名
result = pd.concat([df1, df2], keys=['df1', 'df2'])
# 输出合并后的结果
print(result)
```
输出结果为:
```
A B C D
df1 0 A0 B0 C0 D0
1 A1 B1 C1 D1
2 A2 B2 C2 D2
3 A3 B3 C3 D3
df2 0 A4 B4 C4 D4
1 A5 B5 C5 D5
2 A6 B6 C6 D6
3 A7 B7 C7 D7
```
可以看到,使用 `keys` 参数将原本的列名修改成了 `'df1'` 和 `'df2'`。这样做的好处是,当合并多个 DataFrame 时,可以更方便地区分不同 DataFrame 中相同列名的列。
阅读全文