numpy 花式索引,ix_

#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]]
'''

猜你喜欢

转载自blog.csdn.net/th_num/article/details/80321300
今日推荐