pandas获取列名所在列

目录

判断是否有列；

获取列序号

选取多列数据：

列名转numpy，获取索引：

第一行数据：

---

df_label = pd.read_csv(r"aaa.csv")
    
    df.columns.tolist()

判断是否有列；

if "val/train" in list(df_label)

获取列序号

list(df_label).index("val/train")

选取多列数据：

    df_label = pd.read_csv(r"label.csv")
    index=df_label.columns

    for data in df_label.itertuples():
        data=np.asarray(data[1:5])

列名转numpy，获取索引：

column_np = self.df_label.columns.values

np_index=-1
    if np.where(column_np ==  self.split_col)[0].shape[0] > 0:
        np_index = np.where(column_np == self.split_col)[0][0]
    else:
        pass

第一行数据：

self.cls_1 =self.df_label.iloc[0].tolist()[1:np_index+1]