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