Как мне создать для l oop в Python, чтобы скопировать определенные c позиции в фрейме данных в другой?

Question

У меня есть фрейм данных с именем data6 с 6000 строками, и я хочу скопировать в na 2000 строк данных, называемых result , только значения столбцов месяца, когда значение столбцов уровня равно 1. Как создать для l oop с этим правилом?

Теперь:

in: data6 = df1[['level', 'Month']]
    print(data6)

out:      level   Month
0         1.0  101.52
1         2.0  101.52
2         3.0  101.52
3         1.0  111.89
4         2.0  111.89

Ожидается после l oop:

in: print(result)

out:      level   Month
0         1.0  101.52
1         1.0  111.89
2         1.0  112.27
3         1.0  89.57
4         1.0  110.35

Answer 1

• Использовать логическое индексирование
  • Индексирование и выбор данных

# if level is a float
result = data6[data6.level == 1.0].reset_index(drop=True)

# if level is a string
result = data6[data6.level == '1.0'].reset_index(drop=True)

# if you only want the month column
result = pd.DataFrame(data6.Month[data6.level == 1.0]).reset_index(drop=True)  # or '1.0'