У меня есть файл csv со строками и столбцами, разделенными запятыми. Этот файл содержит заголовки (str) и значения. Теперь я хочу отфильтровать все данные с условием. Например, есть заголовок "pmra", и я хочу сохранить всю информацию для значений pmra между -2,6 и -2,0. Как я могу это сделать? Я пробовал с np.where, но это не сработало. Спасибо за вашу помощь.
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
filename="NGC188_C.csv"
data = pd.read_csv(filename)
ra = data["ra"]
dec = data["dec"]
parallax = data["parallax"]
pm_ra = data["pmra"]
pm_dec = data["pmdec"]
g_band = data["phot_g_mean_mag"]
bp_rp = data["bp_rp"]
Вы можете использовать что-то вроде:
data[(data["pmra"] >= -2.6) & (data["pmra"] <= -2)]
Есть и другой подход: вы можете использовать функцию between
:
data["pmra"].between(-2.6, -2)