有条件地从Python中的矩阵中提取元素。

Question

我对python非常陌生，我想从1911 x 2矩阵中提取一些元素，以便当列1中的元素介于两个值之间时，它们将与列2中的相应元素一起提取到单独的矩阵。不幸的是，我不能只编码矩阵索引，因为矩阵并不总是相同的。

为了让你的背景下面的代码：

import numpy as np 
import pylab 
import matplotlib.pyplot as plt 
from scipy.optimize import leastsq 


data = np.genfromtxt('Ramantest.txt') 

x = data[:,] [:,0] 
y = data[:,] [:,1] 

plt.plot(x,y) 
plt.show() 

的输出中是这样的频谱。
Spectra

我想提取对应于当X值是我想也许使用np.hsplit创建一个新的矩阵2600和2800

之间的情节x和y的值的元素可能是一种选择，但我怎样才能获得代码来确定创建分割的元素？

任何帮助理解

Answer 1

使用np.searchsorted获取数组索引并使用它们切片原始数组。

range = [800,1000] 

idx_low,idx_high = x.searchsorted(data[:,0], range) 

data_within_range = data[idx_low:idx_high,:] 

Answer 2

鉴于任何限制low和high，找到好的索引（low < data[:,0] < high）：

np.logical_and(low < data[:,0], data[:,0] < high) 

然后提取对应于这些索引子阵列：

data[np.logical_and(.low < data[:,0], data[:,0] < high)] 

即使您的x数据没有排序，这也可以工作。

在你的情况下，它看起来像x都是，所以你也能找到的第一个指数i_low和i_high这样data[0, i_low] > low和data[0,i_high] > high（使用searchsorted由Chirstian K.的建议），然后只用一个片： data[i_low:i_high]