import pandas as pd import numpy as np # 读取 Excel 文件 df = pd.read_excel('stockdata.xlsx',parse_dates=['日期']) df['行业代码'] = pd.to_numeric(df['行业代码'], errors='coerce') # 将日期列转换为日期类型,然后提取年份 df.set_index('日期', inplace=True) df.index = pd.to_datetime(df.index) df['年份'] = df.index.year # 使用均值法将季度数据转换为年度数据 #yearly_df = df.groupby(['股票代码', '年份']).mean().reset_index() df = df.groupby(['股票代码', '年份']).mean().reset_index() df[['行业代码','实质控制人性质']] = df['行业代码','实质控制人性质'].str.split(',', expand=True).apply(pd.to_numeric) df = df.groupby(['行业代码','实际控制人性质']).first().reset_index() #arr = np.array(['实际控制人性质']) #first_value = arr[0] # 填充缺失值 df = df.fillna(df.mean()) # 按要求选择需要的列股票代码、年份、实际控制人性质、行业代码、短期负债、长期负债合计、资产负债率 #yearly_df = yearly_df[['股票代码','年份','实际控制人性质','行业代码','短期负债','长期负债','资产负债率']] df = df[['股票代码','年份', '实际控制人性质', '行业代码', '短期负债', '长期负债合计','资产负债率']] # 若一个企业拥有多个实际控制人及行业代码时,以第一个数值为准 #df = df.groupby('行业代码','实际控制人性质').first().reset_index() # 将年份设置为最后一个季度的年份 #df['年份'] = df.groupby('股票代码')['年份'].transform('max') # 将结果写入 Excel 文件 df.to_excel('a2.xlsx', index=False)
时间: 2023-09-03 07:10:06 浏览: 164
在您提供的代码中,有一些语法错误和逻辑错误。以下是经过修改后的代码:
```python
import pandas as pd
import numpy as np
# 读取 Excel 文件
df = pd.read_excel('stockdata.xlsx', parse_dates=['日期'])
df['行业代码'] = pd.to_numeric(df['行业代码'], errors='coerce')
# 将日期列转换为日期类型,然后提取年份
df.set_index('日期', inplace=True)
df.index = pd.to_datetime(df.index)
df['年份'] = df.index.year
# 使用均值法将季度数据转换为年度数据
df = df.groupby(['股票代码', '年份']).mean().reset_index()
# 将实质控制人性质和行业代码列中的逗号分隔的多个数值拆分,并转换为数值类型
df[['行业代码', '实际控制人性质']] = df['行业代码,实际控制人性质'].str.split(',', expand=True).apply(pd.to_numeric)
df.drop(columns=['行业代码,实际控制人性质'], inplace=True)
# 对于一个企业拥有多个实际控制人及行业代码的情况,以第一个数值为准
df = df.groupby(['股票代码', '年份']).first().reset_index()
# 填充缺失值
df = df.fillna(df.mean())
# 选择需要的列股票代码、年份、实际控制人性质、行业代码、短期负债、长期负债合计、资产负债率
df = df[['股票代码', '年份', '实际控制人性质', '行业代码', '短期负债', '长期负债合计', '资产负债率']]
# 将结果写入 Excel 文件
df.to_excel('a2.xlsx', index=False)
```
在这个例子中,我们首先使用 Pandas 库读取原始数据文件“stockdata.xlsx”。然后,根据日期列中的年份进行分组,计算每个分组中其他列的均值,从而将季度数据转化为年度数据。接着,将实质控制人性质和行业代码列中的逗号分隔的多个数值拆分,并转换为数值类型。对于一个企业拥有多个实际控制人及行业代码的情况,以第一个数值为准。使用均值法对缺失值进行填充。最后,选择需要的变量,保存为新的Excel文件“a2.xlsx”。
需要注意的是,实际控制人性质和行业代码的处理是需要根据具体数据情况进行的,我无法得知具体的处理方法。另外,在拆分实质控制人性质和行业代码时,应该使用列名“行业代码,实际控制人性质”,而不是“行业代码”和“实质控制人性质”两个列名。
阅读全文