使用read_csv

Question

我构建了一个矩阵，其中包含列和索引的整数值。矩阵在每个月都是实际的分层。我的问题是，当我将数据写入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

Answer 1

我使用过这样的东西：

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

其中：

df是pandas数据框，列是要更改的列