Выберите фрагменты списка многоиндексного/многостолбцового фрейма данных Pandas ⇐ Python

[Anonymous]

Скажем, у меня есть следующий многостолбцовый DataFrame Pandas:

arrays = [['bar', 'bar', 'baz', 'baz', 'foo', 'foo', ],
['one', 'two', 'one', 'two', 'one', 'two', ]]

tuples = list(zip(*arrays))
index = pd.MultiIndex.from_tuples(tuples, names=['first', 'second'])
df = pd.DataFrame(np.random.randn(8, 6), columns=arrays)

bar baz foo
one two one two one two
0 1.018709 0.295048 -0.735014 1.478292 -0.410116 -0.744684
1 1.388296 0.019284 -1.298793 1.597739 0.044640 -0.040337
2 -0.151763 -0.424984 -1.322985 -0.350483 0.590343 -2.189122
3 -0.221250 -0.449578 -1.512640 0.077380 -0.485380 -0.687565
4 -0.334315 1.790056 0.245414 -0.236784 -0.788226 0.483709
5 -0.943732 1.437968 -0.114556 -1.098798 0.482486 -1.527283
6 -1.213711 1.573547 0.425109 0.513945 0.731550 1.216149
7 0.709976 1.741406 -0.379932 -1.326460 -1.506532 -0.795053


Каков синтаксис выбора комбинации нескольких фрагментов, например выбора ('bar',:) и ('baz':'foo','two')? Я знаю, что могу сделать что-то вроде:

df.loc[:, [('bar', 'one'), ('baz', 'two')]]

bar baz
one two
0 1.018709 1.478292
1 1.388296 1.597739
2 -0.151763 -0.350483
3 -0.221250 0.077380
4 -0.334315 -0.236784
5 -0.943732 -1.098798
6 -1.213711 0.513945
7 0.709976 -1.326460


И что-то вроде:

print(df.loc[:, ('bar', slice(None))])

bar
one two
0 1.018709 0.295048
1 1.388296 0.019284
2 -0.151763 -0.424984
3 -0.221250 -0.449578
4 -0.334315 1.790056
5 -0.943732 1.437968
6 -1.213711 1.573547
7 0.709976 1.741406


Но что-то вроде:

print(df.loc[:, [('bar', slice(None)), ('baz', 'two')]])


Вызывает исключение TypeError, а

print(df.loc[:, ['bar', ('baz', 'two')]])


вызывает исключение ValueError.

Итак, мне нужен простой синтаксис для создания следующего с двумя фрагментами, например:

[('bar', slice(None)), ('baz', 'two')]:

bar baz
one two two
0 -1.438018 1.511736 0.186499
1 -0.432313 -0.478824 -0.055930
2 0.995103 -0.181832 -0.257952
3 0.972293 2.580807 1.536281
4 -0.496261 1.038807 0.209853
5 0.788222 -1.325234 -1.328570


Подробнее здесь: https://stackoverflow.com/questions/428 ... -dataframe