Разделение фрейма данных pandas на основе имени столбца

есть ли способ разделить фрейм данных pandas на основе имени столбца? В качестве примера рассмотрим, что фрейм данных имеет следующие столбцы df = ['A_x', 'B_x', 'C_x', 'A_y', 'B_y', 'C_y'] и я хочу создать два фрейма данных X = ['A_x', 'B_x', 'C_x']и Y = ['A_y', 'B_y', 'C_y'].

Я знаю, что есть возможность сделать это:

d = {'A': df.A_x, 'B': df.B_x, 'C': df.B_x}
X = pd.DataFrame (data=d)

но это было бы не идеально, так как в моем случае у меня есть 2200 столбцов в df. Есть ли более элегантное решение?

1 ответов


можно использовать df.filter(regex=...):

import numpy as np
import pandas as pd
df = pd.DataFrame(np.random.randn(2, 10),
                  columns='Time A_x A_y A_z B_x B_y B_z C_x C_y C-Z'.split())
X = df.filter(regex='_x')
Y = df.filter(regex='_y')

доходность

In [15]: X
Out[15]: 
        A_x       B_x       C_x
0 -0.706589  1.031368 -0.950931
1  0.727826  0.879408 -0.049865

In [16]: Y
Out[16]: 
        A_y       B_y       C_y
0 -0.663647  0.635540 -0.532605
1  0.326718  0.189333 -0.803648