Как запросить dataframe для цикла for с изменяющимися значениями, ValueError: Длина должна совпадать для сравнения?

Question

Это работает, но мне пришлось вводить 4 разных раза для каждого значения в списке charge_names

charge_names = ['Vehicle Theft','Robbery','Burglary','Receive Stolen Property']

charges[charges['Charge Group Description']== 'Vehicle Theft'].head(2)

Я пытался запустить цикл таким образом:

charge_names = ['Vehicle Theft','Robbery','Burglary','Receive Stolen Property']
for name in charge_names:
    charges[charges['Charge Group Description']== name].head(2)

, но не оченьуспех.

это не работает:

charges[['Charge Group Description'].isin(['Robbery', 'Burglary'])]

Как мне запросить все 4 значения в списке charge_names в одной строке?

Answer 1

DataFrame.isin Содержит ли каждый элемент в DataFrame значения.

DataFrame.groupby Группирует DataFrame на основе записей

charge_names = ['Vehicle Theft','Robbery','Burglary','Receive Stolen Property']
charges[charges['Charge Group Description'].isin(charge_names)].groupby('Charge Group Description').head(2)