col = ['rss','intercept'] + ['coef_x_%d'%i for i in range(1,16)]
ind = ['model_pow_%d'%i for i in range(1,16)]
coef_matrix_simple = pd.DataFrame(index=ind, columns=col)

for i in range(1,16):
   coef_matrix_simple.iloc[i-1,0:i+2] = linear_regression(df, power=i, models_to_plot=models_to_plot)

pd.options.display.float_format = '{:,.2g}'.format
coef_matrix_simple

np.savetxt('output.csv', coef_matrix_simple,  delimiter =" ")

Я хочу сохранить coef_matrix_simple вместе со значениями col в качестве имени столбца и значений ind в файле csv. сохранение coef_matrix_simple дает только числа без имени столбца.

1
SANDIPAN DAWN 24 Апр 2018 в 14:27

1 ответ

Лучший ответ

Вы можете попробовать использовать метод pandas DataFrame.to_csv(). Использование в вашем случае будет

coef_matrix_simple.to_csv("output.csv", sep=" ")

Дополнительная информация здесь: https: //pandas.pydata. org / pandas-docs / стабильный / сгенерированный / pandas.DataFrame.to_csv.html

1
Gianluca Micchi 24 Апр 2018 в 12:04