gpt4 book ai didi

python - 如何通过在最后一个元素处拆分数组来将一维数组拆分为 NumPy 中的二维数组?

转载 作者:太空狗 更新时间:2023-10-30 02:55:35 29 4
gpt4 key购买 nike

我有一个 numpy 数组,比方说

 ([1,2,3,4,5,6,7])

我想把它拆分成一个二维数组,这样最后一个元素在它自己的数组中,就像这样

 ([1,2,3,4,5,6],[7])

我该怎么做呢?

最佳答案

使用np.split -

np.split(a,[-1])

sample 运行-

In [105]: a
Out[105]: array([1, 2, 3, 4, 5, 6, 7])

In [106]: np.split(a,[-1])
Out[106]: [array([1, 2, 3, 4, 5, 6]), array([7])]

关于python - 如何通过在最后一个元素处拆分数组来将一维数组拆分为 NumPy 中的二维数组?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42098487/

29 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com