我如何使用 plt.matshow() 從每行隨機 select Numpy 數組的 10 個元素，以及 plot？

Question

幾天來我一直在尋找解決方案，但似乎找不到解決方案。

我有兩個數據集，X 和 Y，X 是 (3000,784)，Y 是 (3000, 1) 二進制 class 數據集。

我如何從 X 的每一行隨機 select 10 個元素，並由此構造一個新矩陣，我可以使用 plt.matshow() 對 Y plot？

我試過使用 np.apply_along_axis() 但我似乎無法正確使用。

   def random_indices(Xtrain, Counts):
       indices = np.random.choice(Xtrain[0,:], Counts, replace=False)
       return indices
Counts = 10

random_indices(Xtrain, Counts)

這是我的 function，它從第一行檢索隨機數，但是我嘗試在所有行上循環它，並嘗試在 np.apply_along_axis 中使用它，但它似乎失敗了

Answer 1

您的數據很小，我想您可以嘗試循環：

def row_sample(X, count):
    return np.array([np.random.choice(row, count, replace=False) for row in X])