сохранение имени столбца переменной в тексте через python

Question

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 дает только цифры без имени столбца.

Answer 1

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

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

Больше информации здесь: https://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.to_csv.html