добавить списки разной длины в фрейм данных pandas

Question

Рассмотрим, у меня есть несколько списков

A = ['acc_num=1', 'A1', 'A2']
B = ['acc_num=2', 'B1', 'B2', 'B3','B4']
C = ['acc_num=3', 'C1']

Как поместить их в массив данных для экспорта в Excel как:

   acc_num     _1  _2  _3  _4  
_1 1           A1  A2
_2 2           B1  B2  B3  B4
_3 3           C1

Answer 1

Привет, это решение для вас в 3 основных этапа c:

1. Создайте DataFrame, просто передав список ваших списков
2. Манипулируйте столбцом acc_num и удалите начальную строку "acc_num=" это делается с помощью строкового метода в векторизованном столбце (но сейчас это может быть далеко)
3. Переименуйте Заголовок / Имена столбцов, как вы, sh, передав словарь {} к df.rename

Код:

# Create a Dataframe from your lists
df = pd.DataFrame([A,B,C])
# Change Column 0 and remove initial string
df[0] = df[0].str.replace('acc_num=','')
# Change the name of Column 0
df.rename(columns={0:"acc_num"},inplace=True)

Окончательный результат:

Out[26]: 
  acc_num   1     2     3     4
0       1  A1    A2  None  None
1       2  B1    B2    B3    B4
2       3  C1  None  None  None