如何用NumPy高效分割列表成等长子列表?

如何用numpy高效分割列表成等长子列表?

NumPy高效分割列表技巧

Python编程中,列表分割是常见操作。当需将列表分割成多个等长子列表时,NumPy提供高效方案。本文讲解如何利用NumPy将长列表分割成多个相同长度的子列表。

问题:如何用NumPy将30个元素的列表分割成3个,每个包含10个元素的子列表?

解决方案:NumPy提供两种方法:reshape和array_split。

方法一:reshape函数

reshape函数可重塑数组形状。由于30能被10整除,可直接用reshape将列表转换成3行10列的二维数组,实现分割。代码如下:

import numpy as npdata = list(range(30))  # 创建包含0-29的列表result = np.array(data).reshape((3, 10))print(result)

登录后复制

本文来自互联网或AI生成,不代表软件指南立场。本站不负任何法律责任。

如若转载请注明出处:http://www.down96.com/tutorials/206.html

热心网友热心网友
上一篇 2025-04-11 13:56
下一篇 2025-04-11 13:57

相关推荐

本站[软件指南]所有内容来自互联网投稿或AI智能生成,并不代表软件指南的立场。