代码之家 › 专栏 › 技术社区 › Denver Dang

删除条目后获取原始列表的索引

numpy python

Denver Dang · 技术社区 · 8 年前

假设我有一个清单:

original_list = [1, 2, 3, 4, 5, 6, 7, 8, 9]

然后我有另一个列表,其中的索引必须从 original_list :

remove_list = [0, 4, 6]

然后使用Numpy删除这些索引:

final_list = np.delete(original_list, remove_list, axis = 0]

很好,假设我现在希望找到 final_list 具有 np.amax(final_list) . 现在有什么方法可以计算出这个值在 原始清单 ?

2 回复 | 直到 8 年前

sacuL 8 年前

选项1: `np.where` :

>>> np.where(original_list == np.amax(final_list))
(array([8]),)

>>> original_list.index(np.amax(final_list))
8

注意:如果有多个匹配索引,此方法将只返回第一个/最小索引。如果需要所有索引,请使用 np.其中 如上所述

Sacry 8 年前

这是另一种方法,

mask = np.zeros(original_list.shape, dtype=np.bool)
mask[remove_list] = True

np.ma.MaskedArray(original_list, mask=mask).argmax()

推荐文章

Redshoe · 当执行numpy.genfromtxt()时,python是否会读取文件的所有行?

1 年前

unfolx · numpy数组不等式的执行时间

1 年前

mchaudh4 · 用numpy表示三对角矩阵

1 年前

Geremia · 2D NumPy数组+1D数组?

1 年前

LMC · Numpy数组布尔索引以获取包含元素

1 年前

HJA24 · 根据条件用值正向填充Numpy矩阵/掩码

1 年前

alpelito7 · 如何只获取numpy数组中每个递增值的第一次出现?

1 年前

Amarth Gûl · 找到一组向量的最近收敛点

1 年前

Mr. W · numpy.divide是函数、类还是其他什么?

1 年前

Mr. W · 为什么numpy.array在编辑内部数据时如此缓慢?

1 年前