多次元リストの作り方を教えてほしいです。

Question

下記のコードのnum_2に格納されている数字を1列ごと違うリストに入れたいです。

実践したこと

import collections
from collections import Counter
from pkgutil import iter_importers
from pprint import pprint

data = [
["A", "B", "C", "D", "E", "E"],
["C", "D", "E", "A", "B", "B"],
["D", "C", "E", "A", "A", "D"],
["D", "A", "C", "B", "D", "E"],
]

items_each_column = []
for c in zip(*data):
    items_each_column.append(len(set(c)))
#print(items_each_column)

frequency = []

for z in zip(*data):
    frequency.append(z)
    z_list = list(z)
   
    nx = list(map(Counter,map(list, zip(*data))))

num = len(nx)


num_2 = []

for i in range(num):
    for j in ["A", "B", "C", "D", "E"]:
        ni = nx[i][j]
        if ni == 0:
            continue
        num_2.append(ni)
print(num_2)

#出力結果
#[1, 1, 2, 1, 1, 1, 1, 2, 2, 2, 1, 1, 1, 1, 1, 1, 1, 1, 2]

このような結果が得られたのですが、想定していた結果は
[[1, 1, 2], [1, 1, 1, 1], [2, 2], [2, 1, 1], [1, 1, 1, 1], [1, 1, 2]]
です。
これはdataの1列目にAが１つ、Cが１つ、Eが2つ、2列目にAが１つ、Bが１つ、Cが１つ、Dが１つ、、、のように対応しています。

Answer 1

内包表記を使いたくなってしまいます・・・

from collections import Counter

data = [
    ["A", "B", "C", "D", "E", "E"],
    ["C", "D", "E", "A", "B", "B"],
    ["D", "C", "E", "A", "A", "D"],
    ["D", "A", "C", "B", "D", "E"],
]

num_2 = [[count[c] for c in "ABCDE" if c in count]
         for count in map(Counter, zip(*data))]
print(num_2)

内包表記を使わないと、以下のようになります。

from collections import Counter

data = [
    ["A", "B", "C", "D", "E", "E"],
    ["C", "D", "E", "A", "B", "B"],
    ["D", "C", "E", "A", "A", "D"],
    ["D", "A", "C", "B", "D", "E"],
]

num_2 = []
for count in map(Counter, zip(*data)):
    num_1 = []
    for c in "ABCDE":
        if c in count:
            num_1.append(count[c])
    num_2.append(num_1)

print(num_2)

Comments

[退会済みユーザー]

すいません。内包表記をまだ理解していないもので、、、でも内包表記の方が短く書けていいなと思っているのでいつかはマスターしたいと思っています。そこで質問なのですが、、、
[[count[c] for c in "ABCDE" if c in count]
for count in map(Counter, zip(*data))]は
for count in map(Counter, zip(*data):
for c in "ABCDE":
if c in count:
print(count[c])
と同じ意味でしょうか？

Answer 2

for i in range(num):
    nj = []
    for j in ["A", "B", "C", "D", "E"]:
        ni = nx[i][j]
        if ni == 0:
            continue
        nj.append(ni)
    num_2.append(nj)
print(num_2)

Comments

[退会済みユーザー]

いつもありがとうございます!!
njというからのリストを作ってその中にniを入れてnum_2というリストに入れるのですね。