Как с помощью фитона выбрать строки, содержащие определенное значение c в фрейме данных? - PullRequest
0 голосов
/ 04 августа 2020

Я новичок в фитоне и программировании, поэтому заранее извините, если я не могу быть настолько ясным. У меня есть фрейм данных, где строки соответствуют идентификаторам (f.ied), а столбцы - нескольким значениям (коды ICD10). Я хочу выбрать строки, содержащие определенные коды c ICD10. Однако я не смог найти правильный способ сделать это ... Я попробовал с lo c и установил, но не повезло ... любая помощь, пожалуйста?

Фрейм данных такой:

каждая строка соответствует f.ied (ID). Я хочу знать, какие f.ied имеют указанные c коды: I20, I21, I22, I23, I24, I25.

Dataframe

1 Ответ

0 голосов
/ 04 августа 2020
df = pd.DataFrame({'feid': [2, 4, 8, 0],
                    'f42002': [2, 0, 0, 0],
                    'f42003': [10, 'I21', 1, 'J10']})

df = df.set_index('feid')
df

DataFrame

    f42002  f42003
feid        
2   2   10
4   0   I21
8   0   1
0   0   J10

Желаемые элементы

mylist = ['I21', 'J10']

for i in mylist:
  print(df[(df['f42002']==i) | (df['f42003']==i)].index.values)

Результат:

[4]
[0]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...