Пример #1
0
 def test_list_indexing(self):
     A = np.arange(6)
     A.shape = (3, 2)
     x = asmatrix(A)
     assert_array_equal(x[:, [1, 0]], x[:, ::-1])
     assert_array_equal(x[[2, 1, 0], :], x[::-1, :])
Пример #2
0
 def test_boolean_indexing(self):
     A = np.arange(6)
     A.shape = (3, 2)
     x = asmatrix(A)
     assert_array_equal(x[:, np.array([True, False])], x[:, 0])
     assert_array_equal(x[np.array([True, False, False]), :], x[0, :])
Пример #3
0
 def test_row_column_indexing(self):
     x = asmatrix(np.eye(2))
     assert_array_equal(x[0, :], [[1, 0]])
     assert_array_equal(x[1, :], [[0, 1]])
     assert_array_equal(x[:, 0], [[1], [0]])
     assert_array_equal(x[:, 1], [[0], [1]])
Пример #4
0
 def test_scalar_indexing(self):
     x = asmatrix(np.zeros((3, 2), float))
     assert_equal(x[0, 0], x[0][0])
Пример #5
0
 def test_basic(self):
     x = asmatrix(np.zeros((3, 2), float))
     y = np.zeros((3, 1), float)
     y[:, 0] = [0.8, 0.2, 0.3]
     x[:, 1] = y > 0.5
     assert_equal(x, [[0, 1], [0, 0], [0, 0]])
Пример #6
0
 def test_asmatrix(self):
     A = np.arange(100).reshape(10, 10)
     mA = asmatrix(A)
     A[0, 0] = -10
     assert_(A[0, 0] == mA[0, 0])