Numpy基础练习(二)

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

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

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

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

vector=np.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=np.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)

[False False False False]

vector=np.array([5,10,15,20])
equal_to_ten_or_five=(vector==10)|(vector==5)
vector[equal_to_ten_or_five]=50
print(vector)

[50 50 15 20]

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

[False  True False]
[[ 5 10 15]
 [20 10 30]
 [35 40 45]]

matrix=np.array(["1","2","3"])
print(vector.dtype)
print(vector)
vector=vector.astype(float)
print(vector.dtype)
print(vector)

int32
[50 50 15 20]
float64
[50. 50. 15. 20.]

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

50

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

array([ 30,  75, 120])

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

array([60, 75, 90])

world_alcohol=np.genfromtxt("world_alcohol.txt",delimiter=",")
is_value_empty=np.isnan(world_alcohol[:,4])
world_alcohol[is_value_empty,4]='0'
alcohol_consumption=world_alcohol[:,4]
alcohol_consumption=alcohol_consumption.astype(float)
total_alcohol=alcohol_consumption.sum()
average_alcohol=alcohol_consumption.mean()
print(total_alcohol)
print(average_alcohol)

1137.78
1.140060120240481