dataframe添加新列
时间: 2023-11-18 20:06:16 浏览: 76
pyspark给dataframe增加新的一列的实现示例
要在DataFrame中添加新列,可以使用以下方法:
1. 使用字典方式添加新列:
```python
import pandas as pd
# 创建DataFrame
df = pd.DataFrame({'name': ['Alice', 'Bob', 'Charlie'],
'age': [25, 30, 35]})
# 添加新列
df['gender'] = ['female', 'male', 'male']
print(df)
```
输出:
```
name age gender
0 Alice 25 female
1 Bob 30 male
2 Charlie 35 male
```
2. 使用apply()方法添加新列:
```python
import pandas as pd
# 创建DataFrame
df = pd.DataFrame({'name': ['Alice', 'Bob', 'Charlie'],
'age': [25, 30, 35]})
# 定义函数
def get_gender(name):
if name == 'Alice':
return 'female'
else:
return 'male'
# 添加新列
df['gender'] = df['name'].apply(get_gender)
print(df)
```
输出:
```
name age gender
0 Alice 25 female
1 Bob 30 male
2 Charlie 35 male
```
以上两种方法都可以用来添加新列,具体取决于你的需求。
阅读全文