pandas將list數據拆分成行或列的實現

更新時間：2020年12月13日 14:40:54 作者：XerCis

這篇文章主要介紹了pandas將list數據拆分成行或列的實現，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

數據

import numpy as np
import pandas as pd

data = [{'Name': '小明', 'Chinese': [70, 80], 'Math': [90, 80]},
    {'Name': '小紅', 'Chinese': [70, 80, 90], 'Math': [90, 80, 70]}]
data = pd.DataFrame(data)
data

在這里插入圖片描述

拆分成行

在這里插入圖片描述

def split_row(data, column):
  '''拆分成行

  :param data: 原始數據
  :param column: 拆分的列名
  :type data: pandas.core.frame.DataFrame
  :type column: str
  '''
  row_len = list(map(len, data[column].values))
  rows = []
  for i in data.columns:
    if i == column:
      row = np.concatenate(data[i].values)
    else:
      row = np.repeat(data[i].values, row_len)
    rows.append(row)
  return pd.DataFrame(np.dstack(tuple(rows))[0], columns=data.columns)


split_row(data, column='Chinese')

拆分成列

在這里插入圖片描述

from copy import deepcopy


def split_col(data, column):
  '''拆分成列

  :param data: 原始數據
  :param column: 拆分的列名
  :type data: pandas.core.frame.DataFrame
  :type column: str
  '''
  data = deepcopy(data)
  max_len = max(list(map(len, data[column].values))) # 最大長度
  new_col = data[column].apply(lambda x: x + [None]*(max_len - len(x))) # 補空值，None可換成np.nan
  new_col = np.array(new_col.tolist()).T # 轉置
  for i, j in enumerate(new_col):
    data[column + str(i)] = j
  return data


split_col(data, column='Chinese')

其他情況

1. 批量處理+不要原列

在這里插入圖片描述

def split_col(data, columns):
  '''拆分成列

  :param data: 原始數據
  :param columns: 拆分的列名
  :type data: pandas.core.frame.DataFrame
  :type columns: list
  '''
  for c in columns:
    new_col = data.pop(c)
    max_len = max(list(map(len, new_col.values))) # 最大長度
    new_col = new_col.apply(lambda x: x + [None]*(max_len - len(x))) # 補空值，None可換成np.nan
    new_col = np.array(new_col.tolist()).T # 轉置
    for i, j in enumerate(new_col):
      data[c + str(i)] = j


split_col(data, columns=['Chinese','Math'])
data

2. 帶int和list數據

在這里插入圖片描述

轉成這樣：

在這里插入圖片描述

import numpy as np
import pandas as pd

data = [{'Name': '小愛', 'Chinese': 70, 'Math': 90},
    {'Name': '小明', 'Chinese': [70, 80], 'Math': [90, 80]},
    {'Name': '小紅', 'Chinese': [70, 80, 90], 'Math': [90, 80, 70]}]
data = pd.DataFrame(data)

def split_col(data, columns):
  '''拆分成列

  :param data: 原始數據
  :param columns: 拆分的列名
  :type data: pandas.core.frame.DataFrame
  :type columns: list
  '''
  for c in columns:
    new_col = data.pop(c)
    max_len = max(list(map(lambda x:len(x) if isinstance(x, list) else 1, new_col.values))) # 最大長度
    new_col = new_col.apply(lambda x: x+[None]*(max_len - len(x)) if isinstance(x, list) else [x]+[None]*(max_len - 1)) # 補空值，None可換成np.nan
    new_col = np.array(new_col.tolist()).T # 轉置
    for i, j in enumerate(new_col):
      data[c + str(i)] = j


split_col(data, columns=['Chinese','Math'])
data