以下是一个示例代码,用于按列写入CSV文件:
import csv
# 定义要写入的数据
data = [
['张三', 18, '男'],
['李四', 20, '女'],
['王五', 22, '男']
]
# 打开CSV文件,以写入模式写入数据
with open('data.csv', 'w', newline='') as file:
writer = csv.writer(file)
# 写入标题行
writer.writerow(['姓名', '年龄', '性别'])
# 按列写入数据
for row in zip(*data):
writer.writerow(row)
上述代码中,首先定义了要写入的数据data
,其中每个元素表示一列数据。然后使用csv.writer
创建一个writer
对象,指定要写入的文件名data.csv
以及打开模式为写入模式。接下来使用writer.writerow
方法写入标题行。最后使用zip(*data)
将data
列表转置,然后使用writer.writerow
方法按列写入数据。
执行以上代码后,会在当前目录下生成一个名为data.csv
的CSV文件,内容如下:
姓名,年龄,性别
张三,李四,王五
18,20,22
男,女,男
注意:由于zip
函数返回的是一个迭代器,所以如果data
列表的长度不一致,会以最短的列表长度为准。如果需要以最长的列表长度为准,可以使用itertools.zip_longest
函数。
上一篇:按列项进行算术运算进行汇总