получить индекс из подмножества pandas multindex [duplicate]

Question

получить индекс из подмножества pandas multindex [duplicate]

1

Учитывая этот образец pandas.core.frame.DataFrame:

Я хотел бы получить те индексы, где col равно 1:

df[df['col']==1]

Ожидаемым результатом будет вектор (0,3).

PeCaDe 20 авг. 2018, в 12:22

Источник

Теги:

python

pandas

numpy

indexing

multi-index

2 ответа

1

Это может сделать трюк:

import pandas as pd

d = {'base':[2,0,3],'col':[0,1,1]}
df = pd.DataFrame(data=d)
col_is_1 = df['col'] == 1
df_temp = df[col_is_1]
matrix = df_temp.values
vec = matrix[:,0]
print(vec)

return [0,3], как ожидалось

Sharku 20 авг. 2018, в 09:01

Ещё вопросы

jpp · Accepted Answer · 2018-08-20T07-15-00.000Z

Вы можете использовать NumPy для создания массива индексов, а затем фильтр для второго индекса:

import numpy as np

df = pd.DataFrame({'col': [0, 1, 1]},
                  index=pd.MultiIndex.from_tuples([(0, 2), (0, 0), (0, 3)]))

#      col
# 0 2    0
#   0    1
#   3    1

idx = np.array(df[df['col'] == 1].index.tolist())

# array([[0, 0],
#        [0, 3]])

res = idx[:, 1]

# array([0, 3])