numpy使用技巧

1.基本结构

1.数组的建立

numbers = np.array([1,2,3,4])#建立数组
print(numbers)
numbers.dtype类型

输出

[1 2 3 4]
dtype('int32')

#数组每个元素的类型都是一样的，int->float->char
2.numpy索引取值，与列表一样
例一

vector = numpy.array([5,10,15,20])
print(vector[0:3])

输出

[ 5 10 15]

例二

matrix = np.array([
            [5,10,15],
            [20,25,30],
            [35,40,45]
            ])
print(matrix[:,1])

输出

[10 25 40]

例三

matrix=np.array([
    [5,10,15],
    [20,25,30],
    [35,40,45]
])
print(matrix[:,0:2])

输出

[[ 5 10]
 [20 25]
 [35 40]]

2.判断
例一

vector = np.array([5,10,15,20])
vector==10

输出

array([False,  True, False, False])

例二

matrix = numpy.array([
    [5,10,15],
    [20,25,30],
    [35,40,45]
])
matrix == 25

输出

array([[False, False, False],
       [False,  True, False],
       [False, False, False]])

例三

vector = numpy.array([5,10,15,20])
equal_to_ten = (vector==10)
print(equal_to_ten)
print(vector[equal_to_ten])

输出

[False  True False False]
[10]

例四

matrix = numpy.array([
    [5,10,15],
    [20,25,30],
    [35,40,45]
])
second_column_25 =(matrix[:,1] == 25)
print(second_column_25)
print(matrix[second_column_25,:])

输出

[False  True False]
[[20 25 30]]

例五（判断与或非）

vector = np.array([5,10,15,20])
equal_to_ten_and_five=(vector == 10) & (vector==5)
print(equal_to_ten_and_five)

equal_to_ten_or_five=(vector==10)|(vector==5)
print(equal_to_ten_or_five)

vector[equal_to_ten_or_five]=50
print(vector)

输出

[False False False False]
[ True  True False False]
[50 50 15 20]

例六（类型转换）

vector = numpy.array(['1','2','3'])
print(vector.dtype)
print(vector)
vector = vector.astype(float)
print(vector.dtype)
print(vector)

例七（按行按列求和）

matrix = numpy.array([
    [5,10,15],
    [20,25,30],
    [35,40,45