pandas基本操作

按行存储数据的二维列表写入数据到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)

 把pymysql获取的结果集写入Exce

import pandas as pd
import pymysql

# 建立数据库连接
connection = pymysql.connect(
    host='localhost',
    user='root',
    password='123456',
    database='eskimo'
)

# 编写SQL查询语句
query = "SELECT * FROM sellorder2021 LIMIT 10000"

# 使用pandas的read_sql函数从数据库中读取数据
df = pd.read_sql(query, connection)

# 关闭数据库连接
connection.close()

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

 获取Excel数据按列存入list

import pandas as pd

df = pd.read_excel(r'C:\Users\Admin\Desktop\数据核对\新建 XLSX 工作表.xlsx', sheet_name='Sheet1')

# 将数据按列写入列表
data_list = []
for column in df.columns:
    column_data = list(df[column])
    column_data.insert(0, column)  # 插入列名到列表首位
    data_list.append(column_data)

print(data_list)

 获取Excel数据按行存入list

import pandas as pd

df = pd.read_excel(r'C:\Users\Admin\Desktop\数据核对\新建 XLSX 工作表.xlsx', sheet_name='Sheet1')
# 将数据按行写入列表
data_list2 = []
column_names = list(df.columns)
column_levels = df.columns.names

data_list2.append(column_names)  # 将列名添加为第一个行数据

for index, row in df.iterrows():
    row_data = list(row)
    data_list2.append(row_data)

print(data_list2)