-
Notifications
You must be signed in to change notification settings - Fork 80
Expand file tree
/
Copy pathnumpy_array_equal_array_equiv.py
More file actions
66 lines (44 loc) · 959 Bytes
/
numpy_array_equal_array_equiv.py
File metadata and controls
66 lines (44 loc) · 959 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import numpy as np
print(np.__version__)
# 1.26.1
a = np.array([0, 1, 2])
b = np.array([0, 1, 2])
c = np.array([3, 4, 5])
print(np.all(a == b))
# True
print(np.all(a == c))
# False
print(np.array_equal(a, b))
# True
print(np.array_equal(a, c))
# False
print(np.array_equiv(a, b))
# True
print(np.array_equiv(a, c))
# False
b_float = b.astype(float)
print(np.array_equal(a, b_float))
# True
print(np.array_equiv(a, b_float))
# True
ones = np.array([1, 1, 1])
print(np.array_equal(ones, 1))
# False
print(np.array_equiv(ones, 1))
# True
a_1d = np.array([0, 1, 2])
a_2d = np.array([[0, 1, 2], [0, 1, 2], [0, 1, 2]])
print(np.array_equal(a_1d, a_2d))
# False
print(np.array_equiv(a_1d, a_2d))
# True
a_nan = np.array([np.nan, 1, 2])
b_nan = np.array([np.nan, 1, 2])
print(np.array_equal(a_nan, b_nan))
# False
print(np.array_equiv(a_nan, b_nan))
# False
print(np.all(a_nan == b_nan))
# False
print(np.array_equal(a_nan, b_nan, True))
# True