如何分解pandas中的选定列

Question

我想实现一个自动为对象赋值的函数。下面是我的代码：

column_list = ['datum', 'parameter_name', 'sample_duration', 'pollutant_standard', 'units_of_measure']
def gnumeric_func(data, columns):
    data[columns] = pd.factorize(data.columns)[0]

data = pd.DataFrame(data, column_list)
gnumeric_func(data, column_list)

怎么了？

Answer 1

使用：

def gnumeric_func (data, columns):
  data[columns] = data[columns].apply(lambda x: pd.factorize(x)[0])
  return data

对于所有列：

def gnumeric_func (data):
  data = data.apply(lambda x: pd.factorize(x)[0])
  return data