从另一个数据帧子切片初始化新数据帧及其列名

Question

 X = dataset.iloc[:, 2:-1].values
 df_X = pd.DataFrame(X)
 df_X.head()

我正在将列切片到数组中指定的索引。有什么办法可以在切片后显示列名而不是显示索引吗？我是熊猫新手。

img

Answer 1

初始化DataFrame时，请指定columns参数。你需要从dataset切出那些标题，就像切出X一样。

X = dataset.iloc[:, 2:-1].values
c = df.columns[2:-1]

df_X = pd.DataFrame(X, columns=c)

Answer 2

1。拆分前保存列：

columns = dataset.columns

第2位。像以前一样拆分列。

3。在拆分df上：

splitted_df.columns = columns