Saya mencoba memilih baris menggunakan .loc dengan satu syarat yang harus dipenuhi beberapa kolom:

# %%
a = (100, 50, 75, 100, 100, 50)
b = (100, 25, 15, 100, 75, 50)
c = (100, 75, 50, 100, 100, 25)

df = pd.DataFrame(zip(a, b, c), columns = ['A' , 'B' , 'C'])

col_lst = ('A' , 'B' , 'C')

Kedua baris "harus" menghasilkan output yang sama:

df.loc[df[col_lst] == 100]
    Out: 
ValueError: Cannot index with multidimensional key

df.loc[(df['A'] == 100) & (df['B'] == 100) & (df['C'] == 100)]
    Out:
    A   B   C
0   100 100 100
3   100 100 100

Apakah tidak mungkin menggunakan .loc dengan daftar label?

2
NCFY 2 Mei 2020, 18:32

1 menjawab

Jawaban Terbaik

Anda perlu mengonversi Tuple menjadi daftar dan menggabungkan dengan all di axis=1:

df[(df[list(col_lst)] == 100).all(1)]
#df.loc[(df[list(col_lst)] == 100).all(1)]
1
anky 2 Mei 2020, 15:36