python拆分list_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 p