pandas写入和读取数据基本操作

发布时间 2023-11-13 22:38:50作者: 悬崖上的金鱼

按行存储数据的二维列表写入数据到Excel

import pandas as pd

# 定义列表数据
data = [
    ['id', '姓名', '国家'],
    [1.0, '曹操', '魏国'],
    [2.0, '刘备', '蜀国'],
    [3.0, '孙权', '吴国'],
    [4.0, '荀彧', '魏国'],
    [5.0, '诸葛亮', '蜀国'],
    [6.0, '周瑜', '吴国'],
    [7.0, '曹仁', '魏国'],
    [8.0, '关羽', '蜀国'],
    [9.0, '吕蒙', '吴国']
]

# 创建DataFrame
df = pd.DataFrame(data[1:3], columns=data[0])

# 将DataFrame写入Excel文件
df.to_excel('output.xlsx', index=False)

按 列存储数据的二维列表写入数据到Excel

import pandas as pd


# 定义数据
data = [
    ['id', 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0],
    ['姓名', '曹操', '刘备', '孙权', '荀彧', '诸葛亮', '周瑜', '曹仁', '关羽', '吕蒙'],
    ['国家', '魏国', '蜀国', '吴国', '魏国', '蜀国', '吴国', '魏国', '蜀国', '吴国']
]

# 转换为DataFrame
df = pd.DataFrame(data)

# 将DataFrame进行转置,即行列转换
df = df.T

# 写入文件
df.to_csv('output.csv', index=False, header=False)