代码之家 › 专栏 › 技术社区 › zerohedge

numpy:使用或将多个指定转换为单个指定

numpy-ndarray indexing numpy arrays python

zerohedge · 技术社区 · 6 年前

taxi_modified 是一个二维数组。

下面的代码是有效的,但似乎不符合python:

taxi_modified[taxi_modified[:, 5] == 2, 15] = 1
taxi_modified[taxi_modified[:, 5] == 3, 15] = 1
taxi_modified[taxi_modified[:, 5] == 5, 15] = 1

如果索引5处的列为2,3, 或 5.

以下操作无效:

taxi_modified[taxi_modified[:, 5] == 2 | 3 | 5, 15] = 1

1 回复 | 直到 6 年前

jpp 6 年前

np.isin (NumPy v1.13+),或 np.in1d 对于旧版本。

下面是一个演示:

# example input array
A = np.arange(16).reshape((4, 4))

# calculate Boolean mask for rows
mask = np.isin(A[:, 1], [1, 5, 13])

# assign values, converting mask to integers
A[np.where(mask), 2] = -1

print(A)

array([[ 0,  1, -1,  3],
       [ 4,  5, -1,  7],
       [ 8,  9, 10, 11],
       [12, 13, -1, 15]])

A[np.where(np.isin(A[:, 1], [1, 5, 13])), 2] = -1

推荐文章

serlingpa · 如何准备我的数据以避免无法推断频率

1 年前

Guillaume · 使用操作从Python列表创建numpy数组

2 年前

user19657580 · 在Python中打印两个numpy数组的列表

2 年前

user19657580 · Python中数组中具有相同元素的索引求和

2 年前

mikanim · 改进二维余弦函数的numpy功能

2 年前

Klimt865 · 在Python中将数组列表转换为列表列表

2 年前

theduker · 计算平均绝对误差时,If语句中赋值前引用的局部变量

2 年前

Lynn · 如果列包含Python中的特定字符串,则从列中删除值

2 年前

JasonX · 运行减法计算

2 年前

Jan Hrubec · 选择numpy数组的前n个元素

2 年前