[Python/파이썬] 3차원 리스트를 map을 이용하여 리스트 안의 합을 구하고 1차원 리스트로 변환하기

* 본 포스팅은 주피터 노트북으로 사용했다.

 

import numpy as np
import pandas as pd

arr10 = [[[26, 22, 12, 12, 22], [87, 11, 13, 17, 19], [12, 23, 34, 23, 17]],
         [[26+1, 22+5, 19, 112, 22], [817, 11, 413, 17, 19], [112, 23, 134, 523, 117]],
         [[26+2, 22+6, 111, 132, 22], [87, 101, 130, 17, 19], [12, 23, 64, 23, 117]],
         [[26+3, 22+7, 12, 121, 22], [87, 171, 163, 17, 19], [12, 23, 634, 23, 167]],
         [[26+4, 22+8, 12, 121, 22], [87, 311, 163, 17, 19], [12, 23, 434, 23, 107]]]
print(arr10)

 

형식을 확인한다.

#3차원 배열
npa = np.array(arr10)
np.shape(npa) #3행5열5장있는것

 

 #내가 푼 결과가 이것과 같아야 하므로 확인차 실행한다.

np.sum(arr10,axis=2)

 

def myF1(i):
    sum=0
    for j in i:
        sum+=j
    return sum

def myF(i):
    return list(map(myF1,i))

list(map(myF,arr10))

 

 

자세한 동작과정을 확인해 보기 위해 print문을 적용해 보았다.