numpy 花式索引，ix_

最新推荐文章于 2025-06-14 12:02:56 发布

原创最新推荐文章于 2025-06-14 12:02:56 发布 · 1.1k 阅读

1 ·

CC 4.0 BY-SA版权

numpy 专栏收录该内容

19 篇文章

订阅专栏

本文通过具体示例展示了如何使用Python的NumPy库进行数组创建、遍历和平铺操作，同时还介绍了如何通过条件表达式修改数组中特定元素的值。

#coding:utf-8

import numpy as np

b=np.arange(12).reshape(3,4)
print(b.flat)
for i in b.flat:
    print(i)

f=b.flat
b.flat=7
print(b)
a=[1,2,3]
b=[4,5,6]
c=(a,b)
print(np.ix_(a,b))

f=np.arange(12)
print(f%4==0)

v=np.array([[1,2,3],[4,5,6]])
print("v:",v)
v[np.ix_(range(2),range(2))]=0
print(v)
'''
输出:
[[0 0 3]
 [0 0 6]]
'''
v2=np.array([[1,2,3],[4,5,6]])
v2[range(2),range(2)]=0

#v[v>1]=0
print(v2)
'''
输出：
[[0 2 3]
 [4 0 6]]
'''
#改变元素的值
v2=np.array([[1,2,3],[4,5,6]])
#v2[range(2),range(2)]=0

v2[(v2>1) & (v2 <5)]=0
print(v2)
'''
输出：
[[1 0 0]
 [0 5 6]]
'''