python - 保存 4D 数组而不丢失其格式

Question

我有这个 4D 数组（numpy.ndarray），我需要以一种在我保存它时它的格式不会改变的方式保存它（因为它应该保持不变），然后在我的 Google Colab 文件中重用它。我尝试以不同的格式保存它，当我上传它并在我的代码中预览它时，即使我将它保存为 .npy 格式，也不再保留以前的格式。我还尝试使用我的 GitHub 存储库中的原始链接导入日期或从我的本地设备上传日期，但仍然没有机会。我会很感激你对这个问题的意见！

进一步阐述：

这是我用来生成 4D 数组的代码：

num1=100000
size = (num1, 8, 4, 4)
prob_0 = 0.3 # 30% of zeros
prob_1 = 1 - prob_0 # 70% of ones
P = np.random.choice([0, 1], size=size, p=[prob_0, prob_1])

我需要将此 P 保存为文件并在 Google Colab 中使用。我必须保存它以维护初始数据，并且每次运行代码时都不会提出不同的 4D 数组。该数据（即P）必须保持不变。

标签： pythonarraysnumpysavegoogle-colaboratory

Answer 1

正常工作np.save_np.load

>>> P.shape
(100000, 8, 4, 4)
>>> np.save("P.npy", P)
>>> P2 = np.load("P.npy")
>>> P2.shape
(100000, 8, 4, 4)
>>> np.allclose(P, P2)
True