在numpy数组和pandas数据帧中选择列的常用方法

您可以在方法中使用if条件,并对pandas数据帧和numpy数组有单独的选择方法。给出下面的示例代码。

def method_1(self, var, col_indices):
    if isinstance(var, pd.DataFrame):
        selected_columns = var[var.columns[col_indices]]
    else:
        selected_columns = var[:,col_indices]

在这里,var是您的输入,可以是numpy数组或pandas dataframe,col_index是要选择的列的索引。