可以使用numpy.rollaxis：

x, y, z = np.rollaxis(A, -1)

假设此输入：

array([[[0, 1, 2],
        [3, 4, 5]]])

输出：

# x, y, z
(array([[0, 3]]), array([[1, 4]]), array([[2, 5]]))

这适用于任何位置，只需指定要用作rollaxis的第二个参数的尺寸：

x, y, z = np.rollaxis(A, 2)

计时：

shape = (7,6,8,2,1,4,5,9,3) ; A = np.arange(np.prod(shape)).reshape(shape)作为输入。

%timeit x, y, z = np.rollaxis(A, -1)
# 1.96 µs ± 59 ns per loop (mean ± std. dev. of 7 runs, 1,000,000 loops each)

%timeit x, y, z = np.moveaxis(A, -1, 0)
# 3.78 µs ± 298 ns per loop (mean ± std. dev. of 7 runs, 100,000 loops each)

# credit to @QuangHoang
%timeit x, y, z = [A[...,i] for i in range(3)]
# 691 ns ± 8.24 ns per loop (mean ± std. dev. of 7 runs, 1,000,000 loops each)