主页 / 问题 / 1546057
Accepted
Владимир
Asked:2023-10-16 01:36:20 +0800 CST2023-10-16 01:36:20 +0800 CST

如何使用numpy在二维数组的每一列中随机选择1个值？

  • 772

如何使用numpy在 2D 数组的每一列中随机选择 1 个值？

鉴于：

array = [
    [1, 2, 3],
    [4, 5, 6], 
    [7, 8, 9]
        ]

例如，输出应如下所示：1,8,6

python

4 个回答

  1. Best Answer
     for i in range(len(a[0])):
     print(a[random.randrange(len(a))][i])
    • 5 
  2. import numpy as np

arr = np.array([
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
])

n, m = arr.shape
print(arr[np.arange(n), np.random.randint(m, size=n)])

    np.arange(n)- 依次取每一列 np.random.randint(m, size=n) -从到 的n随机数0m

    正如他们在评论中所说，我的行/列有错误。

    正确选项：

    arr[np.random.randint(n, size=m), np.arange(m)]
    • 2

  3. 使用 NumPy：

    import numpy as np

a = np.array([
    [1, 2, 3, 30],
    [4, 5, 6, 60],
    [7, 8, 9, 90]
])

n, m = a.shape
b = a[np.random.randint(n, size=m), range(m)]
print(b)

    没有 NumPy：

    import random

a = [
    [1, 2, 3, 30],
    [4, 5, 6, 60],
    [7, 8, 9, 90]
]
# b = [*map(random.choice, zip(*a))]
# b = list(map(random.choice, zip(*a)))
b = [random.choice(c) for c in zip(*a)]
print(b)
    • 2

  4. 任何形状的二维数组的解决方案。将函数应用到列np.random.choice。

    array = np.array([
    [1, 2, 3, 4],
    [4, 5, 6, 18],
    [7, 8, 9, 8]
])
print(np.apply_along_axis(np.random.choice, 0, array))

    [1 2 6 4]

    方阵的解。我们打乱线条并去掉主对角线的元素。

    array = np.array([
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
])
np.random.default_rng().shuffle(array)
print(np.diagonal(array))

    [1 8 6]
    • 1

相关问题