1. ホーム
  2. python

[解決済み] Numpyで配列をN個のチャンクに分割する

2023-05-07 12:17:42

質問

このような リストを均等な大きさのチャンクに分割するにはどうすればよいですか? で、配列をチャンクに分割しています。Numpyを使用して巨大な配列のためにこれをより効率的に行う方法はありますか?

どのように解決するのですか?

試しに numpy.array_split .

ドキュメントから。

>>> x = np.arange(8.0)
>>> np.array_split(x, 3)
    [array([ 0.,  1.,  2.]), array([ 3.,  4.,  5.]), array([ 6.,  7.])]

と同一です。 numpy.split と同じですが、グループの長さが同じでなくても例外を発生させません。

チャンク数 > len(array)の場合、内部にネストされた空白の配列が発生します。 a で保存されている場合は、空の配列を削除することができます。

[x for x in a if x.size > 0]

に保存するだけです。 a に保存してください。