python - 如何在绘制数据之前删除某些值

Question

我是第一次使用Python。我有一个 csv 文件，其中包含几列数据:位置、高度、密度、日期等...我正在绘制高度 (i_h100) v 密度 (i_cd) 并设法使用下面的代码将高度限制为低于 50 的值。我现在想要将 y 轴上的值限制在某个“天”范围内，例如 (85-260)。我不知道如何做到这一点。

import pandas 
import matplotlib.pyplot as plt

data=pandas.read_csv('data.csv')
data.plot(kind='scatter',x='i_h100',y='i_cd')
plt.xlim(right=50)

Answer 1

使用.loc对进入图形的数据进行子集化。

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

# Make some dummy data
np.random.seed(42)
df = pd.DataFrame({'a':np.random.randint(0,365,20),
                   'b':np.random.rand(20),
                   'c':np.random.rand(20)})

# all data: plot of 'b' vs. 'c'
df.plot(kind='scatter', x='b', y='c')
plt.show()

# use .loc to subset data displayed based on value in 'a'
# can also use .loc to restrict values of 'b' displayed rather than plt.xlim
df.loc[df['a'].between(85,260) & (df['b'] < 0.5)].plot(kind='scatter', x='b', y='c')
plt.show()