Python实现一维表与二维表之间的相互转化

article/2025/6/30 5:08:20

对已有数据表进行一维和二维之间的转化：

import pandas as pd# 读入数据：
df = pd.read_excel('2dims.xls','Sheet1')
df

df的结构为：

如上图所示df是一个二维表。

# 将二维数据表转化为一维数据表：
new_data = df.set_index('地区') # 将df中的地区一列设置为索引列
df1 = new_data.stack()  # stack的返回对象df1是一个二级索引Series对象
df2 = df1.reset_index() # 通过reset_index函数将Series对象的二级索引均转化为DataFrame对象的列值
df2.columns = ['地区','时间','金额']
df2

df2的结构为：

# 将一维数据表转化为二维数据表：
s = pd.Series(list(df2['金额']),index=[df2['地区'],df2['时间']])  # 利用一维数据表df2构造一个二级索引Series对象
df3 = s.unstack() # 对s执行stack的逆操作unstack
df3

df3的结构为：

将SQL语句一维表形式的查询结果转化成二维表形式：

说明：在数据库sql50中有一张学生成绩表grade(sid,cid,score)，其中sid表示学生编号，cid表示课程编号，score表示课程成绩。

grade表结构如下图所示：

import pandas as pd
import pymysql# 定义一个上下文管理器：
class DataBase(object):def __init__(self,name,password):# 创建数据库连接self.conn = pymysql.connect('localhost','root',str(password),str(name),charset='utf8')# 创建cursor对象self.cursor = self.conn.cursor()def __enter__(self):return self.cursor  # 返回cursor对象，且这个cursor对象会被赋给with语句中as后面的变量def __exit__(self,exc_type,exc_value,traceback):self.cursor.close()    # 关闭游标对象self.conn.close()      # 断开数据库连接def main():with DataBase('sql50',883721) as db:db.execute('select * from grade')   # 执行sql语句content = db.fetchall()  # 获取数据(db中保存着查询结果集)df = pd.DataFrame(list(content))  # 将从数据库中查询出的数据放入DataFrame对象中return dfif __name__ == '__main__':df = main()new_cols = ['sid','cid','score']
df.columns = new_cols
df.head()

df部分数据如下图所示：

将一维表形式的查询结果转化成二维表形式：

# 将一维数据表转化为二维数据表：
s = pd.Series(list(df['score']),index=[df['sid'],df['cid']])  # 利用一维数据表df2构造一个二级索引Series对象
new_df = s.unstack() # 对s执行stack的逆操作unstack
new_df

new_df的结构为：