pd.saveto 和 pd.read_csv 添加標題和索引列

Question

我有一個數據框，我像這樣保存：

Y_train_1.head()

0    4691.0
1    4661.0
2    4631.0
3    4601.0
4    4571.0

Y_train_1.to_csv("Y_train.csv", sep='\t', decimal=',')

然后我像這樣再次閱讀：

Y_train = pd.read_csv("Y_train.csv", sep='\t', decimal=',', index_col=None, header=None))
Y_train.head()

使用

Y_train = pd.read_csv("Y_train.csv", sep='\t', decimal=',', index_col=[0], header=None)

我仍然得到這樣的數據框，這與以前不一樣？

我怎樣才能獲得與以前相同的數據幀？

使用

謝謝，R

Answer 1

如果使用Series添加參數squeeze=True ：

print (type(Y_train_1))
<class 'pandas.core.series.Series'>

print (Y_train_1)
0    4691.0
1    4661.0
2    4631.0
3    4601.0
4    4571.0
dtype: float64

Y_train_1.to_csv("Y_train.csv", sep='\t', decimal=',', header=None)
Y_train = pd.read_csv("Y_train.csv", 
                      sep='\t', 
                      decimal=',', 
                      index_col=[0], 
                      squeeze=True, 
                      header=None)
print (Y_train)
0
0    4691.0
1    4661.0
2    4631.0
3    4601.0
4    4571.0
Name: 1, dtype: float64

---

Y_train_1.to_csv("Y_train.csv", sep='\t', decimal=',', index=False, header=None)
Y_train = pd.read_csv("Y_train.csv", sep='\t', decimal=',', squeeze=True, header=None)
print (Y_train)
0    4691.0
1    4661.0
2    4631.0
3    4601.0
4    4571.0
Name: 0, dtype: float64