18 lines
694 B
Python
18 lines
694 B
Python
|
import pandas as pd
|
||
|
|
|
||
|
|
Firm_Prod = pd.read_csv("input_data/Firm_amended.csv")
|
||
|
|
Firm_Prod.fillna(0, inplace=True)
|
||
|
|
|
||
|
|
# Stack the firm-product relationships into a DataFrame
|
||
|
|
firm_prod = pd.DataFrame({'bool': Firm_Prod.loc[:, '1':].stack()})
|
||
|
|
firm_prod = firm_prod[firm_prod['bool'] == 1].reset_index()
|
||
|
|
firm_prod.drop('bool', axis=1, inplace=True)
|
||
|
|
firm_prod.rename({'level_0': 'Firm_Code', 'level_1': 'Product_Code'}, axis=1, inplace=True)
|
||
|
|
firm_prod['Firm_Code'] = firm_prod['Firm_Code'].astype('string')
|
||
|
|
|
||
|
|
# 保存为新的 CSV 文件
|
||
|
|
output_file_path = 'input_data/firm_industry_relation.csv'
|
||
|
|
firm_prod.to_csv(output_file_path, index=False)
|
||
|
|
|
||
|
|
print(f"新的 CSV 文件已保存到: {output_file_path}")
|