X = dataset.iloc[:, 2:-1].values
df_X = pd.DataFrame(X)
df_X.head()
我正在将列切片到数组中指定的索引。有什么办法可以在切片后显示列名而不是显示索引吗?我是熊猫新手。
答案 0 :(得分:0)
第一。在拆分前保存列:
columns = dataset.columns
第二。像以前一样拆分列。
第三。在拆分的df上执行:
splitted_df.columns = columns
答案 1 :(得分:0)
初始化DataFrame时,请指定columns
参数。您需要从dataset
切出这些标题,方法与切出X
的方式相同。
X = dataset.iloc[:, 2:-1].values
c = df.columns[2:-1]
df_X = pd.DataFrame(X, columns=c)