Добавить строку в список, если значение истинно

Я пытаюсь добавить любую строку, где Ans == True, в отдельный список. Используя ниже, я получаю ошибку типа. В идеале есть более эффективный способ, но мне нужно перебирать строку за строкой.

import pandas as pd

df = pd.DataFrame({
   'Code': ['t1', 't2','t3'],
   'Ans': [True, False, True],
   'Val': [140, 145,140],
})

Trues = []

for row in df.iterrows():
    if row['Ans'] == True:
        Trues.append(row)

print(Trues)

Вне:

TypeError: tuple indices must be integers or slices, not str

Я думаю, вы хотели сказать if row['Ans'] == True вместо if df['Ans'] == True.

John Gordon 31.05.2024 03:38

Просто df = df[df["Ans"]], чтобы выбрать строки, в которых это правда

Kraigolas 31.05.2024 03:42

@JohnGordon, извини, ошибся с концом

tonydanza123 31.05.2024 03:45
.iterrows() возвращает кортеж (index, row). Попробуйте for index, row in df.iterrows():. Если вам действительно нужно повторить.
Amadan 31.05.2024 03:48
Почему в Python есть оператор "pass"?
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Некоторые методы, о которых вы не знали, что они существуют в Python
Некоторые методы, о которых вы не знали, что они существуют в Python
Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...
Основы Python Часть I
Основы Python Часть I
Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:
Оптимизация кода с помощью тернарного оператора Python
Оптимизация кода с помощью тернарного оператора Python
И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это
Советы по эффективной веб-разработке с помощью Python
Советы по эффективной веб-разработке с помощью Python
Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.
1
4
60
1
Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Вы не должны добавлять «строки» в список. .iterrows перебирает «строки» как pd.Series объекты, и нет смысла создавать их список. Вам почти наверняка следует просто подстроить свой фрейм данных, используя сам логический столбец:

In [2]: import pandas as pd

In [3]: df = pd.DataFrame({
   ...:    'Code': ['t1', 't2','t3'],
   ...:    'Ans': [True, False, True],
   ...:    'Val': [140, 145,140],
   ...: })

In [4]: df_true = df[df['Ans']]

In [5]: df_true
Out[5]:
  Code   Ans  Val
0   t1  True  140
2   t3  True  140

Если вам необходим list, то получите list из dict, используя встроенные методы:

In [6]: mylist = df[df['Ans']].to_dict(orient='records')

In [7]: mylist
Out[7]:
[{'Code': 't1', 'Ans': True, 'Val': 140},
 {'Code': 't3', 'Ans': True, 'Val': 140}]

Проблема вашего подхода в том, что .iterrows перебирает пары индексов и строк. Поэтому просто игнорируйте первый элемент пары при итерации, назначая его обычной выбрасываемой переменной _:

In [8]: [row for _, row in df.iterrows() if row['Ans']]
Out[8]:
[Code      t1
 Ans     True
 Val      140
 Name: 0, dtype: object,
 Code      t3
 Ans     True
 Val      140
 Name: 2, dtype: object]

Другие вопросы по теме