python用pandas写入excel时怎么添加标记列
时间: 2023-10-09 13:07:07 浏览: 77
Python pandas如何向excel添加数据
可以使用pandas中的`DataFrame`对象的`apply`方法将标记列添加到数据中,然后使用`to_excel`方法将数据写入Excel文件。以下是示例代码:
```python
import pandas as pd
# 创建示例数据
data = {
'Name': ['Alice', 'Bob', 'Charlie', 'David'],
'Age': [25, 30, 35, 40],
'Gender': ['F', 'M', 'M', 'M']
}
df = pd.DataFrame(data)
# 添加标记列
df['Flag'] = df.apply(lambda row: 'Y' if row['Age'] > 30 else 'N', axis=1)
# 将数据写入Excel文件
writer = pd.ExcelWriter('output.xlsx')
df.to_excel(writer, index=False)
writer.save()
```
在上述示例中,`apply`方法将在每行数据中执行,对于每行数据,如果`Age`列的值大于30,则将标记列设置为`Y`,否则设置为`N`。最后,数据和标记列将写入Excel文件`output.xlsx`中。
阅读全文