提取数组中的每个第3个数据

时间:2013-11-06 03:12:35

标签: python graph numpy

我有数千个x和y数据,在这种情况下我将只使用12个数据。 该数组用于绘制图表

x = np.array([1000,2000,3000,4000,5000,6000,7000,8000,9000,10000,11000,12000])
y = np.array([1,2,3,4,5,6,7,8,9,10,11,12])
py.plot(x,y)

我如何为每个[3]乘法提取绘图?例如

x = np.array([3000,6000,9000,12000])
y = np.array([3,6,9,12])
py.plot(x,y)

我如何为每个[5]乘法提取绘图?例如

x = np.array([5000,10000])
y = np.array([5,10])
py.plot(x,y)

4 个答案:

答案 0 :(得分:3)

从第三个项目(一维数组)开始提取每三个项目

x[2::3], y[2::3]

从第五项(一维数组)

开始提取每五个项目
x[4::5], y[4::5]

答案 1 :(得分:2)

如果您只是询问如何提取每个第x个项目,切片会接受步骤参数以及开始和停止。

例如:

In [1]: import numpy as np

In [2]: np.arange(10)
Out[2]: array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

In [3]: x = np.arange(10)

In [4]: x[::2]
Out[4]: array([0, 2, 4, 6, 8])

In [5]: x[::3]
Out[5]: array([0, 3, 6, 9])

In [6]: x[3::3]
Out[6]: array([3, 6, 9])

如果你问如何找到偶数倍数,你可以使用布尔索引。

例如:

In [7]: x[x % 3 == 0]
Out[7]: array([0, 3, 6, 9])

答案 2 :(得分:2)

使用numpy.where

>>> x = np.array([1000,2000,3000,4000,5000,6000,7000,8000,9000,10000,11000,12000])
>>> y = np.array([1,2,3,4,5,6,7,8,9,10,11,12])
>>> idxs = np.where(x % 3 == 0)
>>> x[idxs]
array([ 3000,  6000,  9000, 12000])
>>> y[idxs]
array([ 3,  6,  9, 12])

>>> x[x % 3 == 0]
array([ 3000,  6000,  9000, 12000])

答案 3 :(得分:0)

使用返回限定值的函数。例如:

def filter_indices(lst, value):
    return [v for _, v in enumerate(lst) if v % value == 0]

结果:

>>> filter_indices(l, 3)
[3, 6, 9, 12]

或者,使用np.where功能:

indices = np.where(x % 3 == 0)
py.plot(x[indices], y[indices])
相关问题