Обновление столбцов в списке данных панд

Question

У меня есть список данных панд. Я хотел бы отфильтровать столбцы в каждом цикле данных в цикле for.

import pandas as pd
df_list=[]
sales = [('Jones LLC', 150, 200, 50),
        ('Alpha Co', 200, 210, 90),
        ('Blue Inc', 140, 215, 95)]
labels = ['account', 'Jan', 'Feb', 'Mar']

df_list.append(pd.DataFrame.from_records(sales, columns=labels))
sales = [('Jones LLC', 122, 566, 345),
        ('Alpha Co', 200, 210, 652),
        ('Blue Inc', 140, 215, 788)]
labels = ['account', 'Jan', 'Feb', 'Mar']
df_list.append(pd.DataFrame.from_records(sales, columns=labels))

col_list = [col for col in df_list[0].columns if col.startswith('acc') or col.startswith('mar')]

#this works
df_list[0]=df_list[0][col_list]
print(df_list[0])

#this does not work
for frame in df_list:
    frame = frame[col_list]
print(df_list[1])

Почему не работает цикл?

Answer 1

Поскольку ваш цикл на самом деле ничего не переопределяет в df_list, он просто повторяется. Вы можете сделать:

for i, frame in enumerate(df_list):
    df_list[i] = frame[col_list]

Который работает:

>>> df_list[0]
     account
0  Jones LLC
1   Alpha Co
2   Blue Inc
>>> df_list[1]
     account
0  Jones LLC
1   Alpha Co
2   Blue Inc