3

リストの配列があるとします

B = [[1,2,3],[1,2,3,4],[1,2]]

そして、列全体のリスト内の要素の平均を計算したいと思います。どうすればいいですか？

すなわち

最長のリストに等しい平均値の最終的な配列を取得するにはどうすればよいですか:

[(1+1+1)/3,(2+2+2)/3,(3+3)/2,4/1] = [1,2,3,4]

私は試した：

final_array = np.array([mean(a) for a in zip(*(B))])

しかし、これは私の最短リストと同じ長さの配列しか与えません。これはマスクが役に立ちましたか？リストの配列にうんざりしていたら申し訳ありませんが、私はまだ Python に慣れています。

4

6 に答える 6

6

pandas の DataFrame を使用してそれを行うことができます。

from pandas import DataFrame

B = [[1,2,3],[1,2,3,4],[1,2]]
df = DataFrame(B)
df.mean(axis=0)
""""
df
   0  1   2   3
0  1  2   3 NaN
1  1  2   3   4
2  1  2 NaN NaN

df.mean(axis=0)
0    1
1    2
2    3
3    4
"""
于 2013-06-11T12:49:25.733 に答える
4

リストにセンチネル値 (私は NaN を使用) を埋め込んでから、そのセンチネルでマスクされた配列を作成する必要があります。マスクされた配列を取得したら、問題なく平均を計算できます。

>>> import numpy as np
>>> B = [[1,2,3],[1,2,3,4],[1,2]]
>>> 
>>> maxlen = max(len(x) for x in B)
>>> C = np.array([l+[np.nan]*(maxlen-len(l)) for l in B])
>>> C
array([[  1.,   2.,   3.,  nan],
       [  1.,   2.,   3.,   4.],
       [  1.,   2.,  nan,  nan]])
>>> dat = np.ma.fix_invalid(C)
>>> np.mean(dat,axis=0)
masked_array(data = [1.0 2.0 3.0 4.0],
             mask = [False False False False],
       fill_value = 1e+20)
于 2013-06-11T12:40:27.497 に答える
1

cmpとを使用するもう 1 つの方法izip_longest

from itertools import izip_longest
[float(sum(col)) / sum(cmp(x,0) for x in col) for col in izip_longest(*B, fillvalue=0)]

これは、値が正であることを前提としています。

于 2013-06-11T13:11:23.423 に答える
0 
B = [[1,2,3],[1,2,3,4],[1,2]]
data = {}
max_len = 0

for alist in B:
    length = len(alist)
    max_len = length if (length > max_len) else max_len

    for i in range(length):
        data.setdefault(i, []).append(alist[i])


results = []

for i in range(max_len):
    vals = data[i]
    results.append(sum(vals) / len(vals) )

print results

--output:--
[1, 2, 3, 4]
于 2013-06-11T12:49:00.183 に答える
0

外部ライブラリなしで実行できます：

B = [[1,2,3],[1,2,3,4],[1,2]]
#compute max length of sub list
maxLen = max([len(x) for x in B])
#new list with number of empty lists equals to number of columns
transList = [[] for i in range(maxLen)]
#transforming list to new structure
for row in B:
    for col in row:
        transList[col-1].append(col)
#transList = [[1, 1, 1], [2, 2, 2], [3, 3], [4]] from now one its simple to get mean of the elements ;)
meanB = [float(sum(i))/len(i) for i in transList]
于 2013-06-11T13:05:52.597 に答える