首页 文章

使用read_csv将列名转换为整数

提问于
浏览
0

我构建了一个矩阵,其中包含列和索引的整数值 . 矩阵在每个月都是实际的分层 . 我的问题是,当我将数据写入csv然后加载为pandas数据帧时,索引和数据选择不再像以前那样有效 .

在写入和读取数据到文件之前选择数据:

matrix.ix[1][4][3] 例如会给 123

在单词select,month January中,让我从原点4到目的地3的(旅行)流程 .

在将数据写入csv并将其读回pandas后,原始引用失败,但如果我将列索引转换为字符串,则可以正常工作:

matrix.ix[1]['4'][3]

...列名已自动从整数转换为字符串 . 但我更喜欢原始索引 . 有什么建议?

我目前从csv加载后处理数据的快速修复方法是:

#Writing df to file
mulitindex_df_Travel_monthly.to_csv(r'result/Final_monthly_FlightData_countrylevel_v4.csv')


#Loading df from csv
test_matrix = pd.read_csv(filepath_inputdata+'/Final_monthly_FlightData_countrylevel_v4.csv', 
                                       index_col=[0, 1])


test_matrix.rename(columns = int, inplace = True) #Thx, @ayhan

CSV文件:https://www.dropbox.com/s/4u2opzh65zwcn81/travel_matrix_SO.csv?dl=0

example df

1 回答

  • 0

    我用过这样的东西:

    df = df.rename(columns={str(c): c for c in columns})

    哪里:

    df是pandas数据帧,列是要更改的列

相关问题