web-dev-qa-db-ja.com

リストを小さなリストに分割(半分に分割)

pythonリストを簡単に半分に分割する方法を探しています。

配列がある場合:

A = [0,1,2,3,4,5]

私は得ることができるだろう:

B = [0,1,2]

C = [3,4,5]
121
corymathews
A = [1,2,3,4,5,6]
B = A[:len(A)//2]
C = A[len(A)//2:]

機能が必要な場合:

def split_list(a_list):
    half = len(a_list)//2
    return a_list[:half], a_list[half:]

A = [1,2,3,4,5,6]
B, C = split_list(A)
179
Jason Coon

もう少し一般的な解決策(単に半分に分割するのではなく、必要な部分の数を指定できます):

EDIT:奇数リストの長さを処理するために投稿を更新しました

EDIT2:Briansの有益なコメントに基づいて投稿を再度更新する

def split_list(alist, wanted_parts=1):
    length = len(alist)
    return [ alist[i*length // wanted_parts: (i+1)*length // wanted_parts] 
             for i in range(wanted_parts) ]

A = [0,1,2,3,4,5,6,7,8,9]

print split_list(A, wanted_parts=1)
print split_list(A, wanted_parts=2)
print split_list(A, wanted_parts=8)
73
ChristopheD
f = lambda A, n=3: [A[i:i+n] for i in range(0, len(A), n)]
f(A)

n-結果配列の事前定義された長さ

38
Jaro
def split(arr, size):
     arrs = []
     while len(arr) > size:
         pice = arr[:size]
         arrs.append(pice)
         arr   = arr[size:]
     arrs.append(arr)
     return arrs

テスト:

x=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]
print(split(x, 5))

結果:

[[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11, 12, 13]]
28
Siamand Maroufi

B,C=A[:len(A)/2],A[len(A)/2:]

12
John Montgomery

ここに一般的な解決策があり、arrをカウント部分に分割

def split(arr, count):
     return [arr[i::count] for i in range(count)]
11
Chris Song

注文を気にしない場合...

def split(list):  
    return list[::2], list[1::2]

list[::2]は、リストの2番目の要素を0番目の要素から取得します。
list[1::2]は、リストの2番目の要素を1番目の要素から取得します。

10
sentythee
def splitter(A):
    B = A[0:len(A)//2]
    C = A[len(A)//2:]

 return (B,C)

テストしたところ、python 3でint除算を強制するにはダブルスラッシュが必要です。元の投稿は正しいものでしたが、何らかの理由でOperaでwysiwygが壊れました。

7
Stefan Kendall

配列をサイズnのより小さな配列に分割する、より一般化された場合の公式Pythonレシピがあります。

from itertools import izip_longest
def grouper(n, iterable, fillvalue=None):
    "Collect data into fixed-length chunks or blocks"
    # grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx
    args = [iter(iterable)] * n
    return izip_longest(fillvalue=fillvalue, *args)

このコードスニペットは python itertools doc page からのものです。

6
Jonathan Berger

リストのスライス を使用します。構文は基本的にmy_list[start_index:end_index]です

>>> i = [0,1,2,3,4,5]
>>> i[:3] # same as i[0:3] - grabs from first to third index (0->2)
[0, 1, 2]
>>> i[3:] # same as i[3:len(i)] - grabs from fourth index to end
[3, 4, 5]

リストの前半を取得するには、最初のインデックスからlen(i)//2にスライスします(ここで//は整数除算です-したがって3//2 will give the floored result of1, instead of the invalid list index of1.5`):

>>> i[:len(i)//2]
[0, 1, 2]

..そして値を入れ替えて後半を取得します:

>>> i[len(i)//2:]
[3, 4, 5]
4
dbr

上記の答えは多かれ少なかれ正しいですが、配列のサイズが2で割り切れない場合は、python 3.0でa / 2の結果が浮動小数点数になるため、問題が発生する可能性があります。以前のバージョンでは、スクリプトの先頭でfrom __future__ import divisionを指定した場合。いずれにしても、コードの「前方」互換性を得るために、整数除算、つまりa // 2を使用した方が良いでしょう。

2
user91259

大きなリストがある場合は、itertoolsを使用し、必要に応じて各部分を生成する関数を作成することをお勧めします。

from itertools import islice

def make_chunks(data, SIZE):
    it = iter(data)
    # use `xragne` if you are in python 2.7:
    for i in range(0, len(data), SIZE):
        yield [k for k in islice(it, SIZE)]

次のように使用できます。

A = [0, 1, 2, 3, 4, 5, 6]

size = len(A) // 2

for sample in make_chunks(A, size):
    print(sample)

出力は次のとおりです。

[0, 1, 2]
[3, 4, 5]
[6]

@ thefourtheye および @ Bede Constantinides に感謝

2

10年後。

arr = 'Some random string' * 10; n = 4
print([arr[e:e+n] for e in range(0,len(arr),n)])
2
RoyM

これは他のソリューションと似ていますが、少し高速です。

# Usage: split_half([1,2,3,4,5]) Result: ([1, 2], [3, 4, 5])

def split_half(a):
    half = len(a) >> 1
    return a[:half], a[half:]

@ChristopheDからのヒント付き

def line_split(N, K=1):
    length = len(N)
    return [N[i*length/K:(i+1)*length/K] for i in range(K)]

A = [0,1,2,3,4,5,6,7,8,9]
print line_split(A,1)
print line_split(A,2)
0
PunjCoder
#for python 3
    A = [0,1,2,3,4,5]
    l = len(A)/2
    B = A[:int(l)]
    C = A[int(l):]       
0
SuperGuy10