почему в pandas уравнениях не используются значения в столбцах

Question

У меня есть фрейм данных с числами, и они распечатываются с помощью команды печати, поэтому я знаю, что он находится в фрейме данных. Но когда я делаю свои уравнения и условные выражения, их нет в переменных.

import pandas as pd
data = pd.read_excel('Cam_practice1.xlsx')
df = pd.DataFrame(data, columns = ['x_block', 'y_block'])

print(df)
equation_x = ((df.x_block))**2
equation_y = ((df.y_block))**2
eq = equation_x + equation_y

if eq  <=4 :
    df.to_csv('gridoutput.csv')

Я хочу использовать полную формулу eq, когда это значение меньше или равно 4, я хочу, чтобы строка была записано на новый вывод. Где я ошибаюсь?

Answer 1

Вы можете это сделать:

equation_x = ((df.x_block))**2
equation_y = ((df.y_block))**2
eq = equation_x + equation_y
df[eq<=4].to_csv('gridoutput.csv')