Python: スプレッドシートで各要素の掛け合わせの表を作る

次のような Python ディクショナリ型の値をスプレッドシートに表の形で記述することを考えます。

pattern = {
'エリア': ['日本', 'アメリカ', 'イギリス'],
'性別': ['男', '女'],
'身長': ['100cm以上', '100cm未満']
}

pattern = {

'エリア': ['日本', 'アメリカ', 'イギリス'],

'性別': ['男', '女'],

'身長': ['100cm以上', '100cm未満']

}

環境

Python 3.7.6
Linux

掛け合わせ表の作成 (直積)

スプレッドシートで次のような各要素を掛け合わせた表を作ります。

エリア	性別	身長
1	日本	男	100cm以上
2	日本	男	100cm未満
3	日本	女	100cm以上
4	日本	女	100cm未満
5	アメリカ	男	100cm以上
6	アメリカ	男	100cm未満
7	アメリカ	女	100cm以上
8	アメリカ	女	100cm未満
9	イギリス	男	100cm以上
10	イギリス	男	100cm未満
11	イギリス	女	100cm以上
12	イギリス	女	100cm未満

Python で次のように書くと、カンマ区切りのものが出力されます。

from itertools import product
if __name__ == '__main__':
pattern = {
'エリア': ['日本', 'アメリカ', 'イギリス'],
'性別': ['男', '女'],
'身長': ['100cm以上', '100cm未満']
}
delimiter = ','
result = delimiter + delimiter.join(pattern.keys()) + '\n'
i = 0
for item in product(*pattern.values()):
i += 1
result += str(i) + delimiter + delimiter.join(item) + '\n'
print(result)

from itertools import product

if __name__ == '__main__':

pattern = {

'エリア': ['日本', 'アメリカ', 'イギリス'],

'性別': ['男', '女'],

'身長': ['100cm以上', '100cm未満']

}

delimiter = ','

result = delimiter + delimiter.join(pattern.keys()) + '\n'

i = 0

for item in product(*pattern.values()):

i += 1

result += str(i) + delimiter + delimiter.join(item) + '\n'

print(result)

それをスプレッドシートに貼り付けるとうまい具合に表ができます。

,エリア,性別,身長
1,日本,男,100cm以上
2,日本,男,100cm未満
3,日本,女,100cm以上
4,日本,女,100cm未満
5,アメリカ,男,100cm以上
6,アメリカ,男,100cm未満
7,アメリカ,女,100cm以上
8,アメリカ,女,100cm未満
9,イギリス,男,100cm以上
10,イギリス,男,100cm未満
11,イギリス,女,100cm以上
12,イギリス,女,100cm未満

,エリア,性別,身長

1,日本,男,100cm以上

2,日本,男,100cm未満

3,日本,女,100cm以上

4,日本,女,100cm未満

5,アメリカ,男,100cm以上

6,アメリカ,男,100cm未満

7,アメリカ,女,100cm以上

8,アメリカ,女,100cm未満

9,イギリス,男,100cm以上

10,イギリス,男,100cm未満

11,イギリス,女,100cm以上

12,イギリス,女,100cm未満

補足

itertools.product を使って、複数のリストの掛け合わせを作っています。この関数は可変長引数ですから、 *pattern.values() と書いてリストを展開する書き方にしています。
for の中で使う i は emumerate を使って作ることもできます。次のコードでは enumerate を使っています。

分類項目表の作成

上の掛け合わせ表の各項目とその値について次のような表を作りたい場合

項目	値
エリア	日本
	アメリカ
	イギリス
性別	男
	女
身長	100cm以上
	100cm未満

Python で次のように書くと、カンマ区切りのものが出力されます。

if __name__ == '__main__':
pattern = {
'エリア': ['日本', 'アメリカ', 'イギリス'],
'性別': ['男', '女'],
'身長': ['100cm以上', '100cm未満']
}
delimiter = ','
result = ''
for key, value in pattern.items():
for index, item in enumerate(value):
if index == 0:
result += key
result += delimiter + item + '\n'
print(result)

if __name__ == '__main__':

pattern = {

'エリア': ['日本', 'アメリカ', 'イギリス'],

'性別': ['男', '女'],

'身長': ['100cm以上', '100cm未満']

}

delimiter = ','

result = ''

for key, value in pattern.items():

for index, item in enumerate(value):

if index == 0:

result += key

result += delimiter + item + '\n'

print(result)

それをスプレッドシートに貼り付けるとうまい具合に表ができます。

エリア,日本
,アメリカ
,イギリス
性別,男
,女
身長,100cm以上
,100cm未満

エリア,日本

,アメリカ

,イギリス

性別,男

,女

身長,100cm以上

,100cm未満

まとめたコード

これを対話型で進められるひとつのPythonファイルにすると次のようになります。

from itertools import product
def get_input(default: str = ''):
value: str = input().rstrip("\n")
if value:
return value
return default
def configure():
print("""
Please configure
delimiter [,]:
""")
delimiter = get_input(',')
print(f"""
prefix:
""")
prefix = get_input()
print(f"""
suffix:
""")
suffix = get_input()
return delimiter, prefix, suffix
def print_product_table(pattern: dict, delimiter: str, prefix: str, suffix: str):
result = prefix + delimiter + delimiter.join(pattern.keys()) + suffix + '\n'
i = 0
for item in product(*pattern.values()):
i += 1
result += prefix + str(i) + delimiter + delimiter.join(item) + suffix + '\n'
print(result)
def print_category_key_value(pattern, delimiter, prefix, suffix):
result = ''
for key, value in pattern.items():
for index, item in enumerate(value):
line = prefix
if index == 0:
line += key
line += delimiter + item + suffix
result += line.strip() + '\n'
print(result)
if __name__ == '__main__':
pattern = {
'エリア': ['日本', 'アメリカ', 'イギリス'],
'性別': ['男', '女'],
'身長': ['100cm以上', '100cm未満']
}
print_pattern: int = None
while True:
print("""
Print dict structure.
Enter print pattern:
1. Product table
2. Category key-value
""")
print_pattern = int(get_input())
if print_pattern in [1, 2]:
break;
delimiter, prefix, suffix = configure()
if print_pattern == 1:
print_product_table(pattern, delimiter, prefix, suffix)
elif print_pattern == 2:
print_category_key_value(pattern, delimiter, prefix, suffix)

from itertools import product

def get_input(default: str = ''):

value: str = input().rstrip("\n")

if value:

return value

return default

def configure():

print("""

Please configure

delimiter [,]:

""")

delimiter = get_input(',')

print(f"""

prefix:

""")

prefix = get_input()

print(f"""

suffix:

""")

suffix = get_input()

return delimiter, prefix, suffix

def print_product_table(pattern: dict, delimiter: str, prefix: str, suffix: str):

result = prefix + delimiter + delimiter.join(pattern.keys()) + suffix + '\n'

i = 0

for item in product(*pattern.values()):

i += 1

result += prefix + str(i) + delimiter + delimiter.join(item) + suffix + '\n'

print(result)

def print_category_key_value(pattern, delimiter, prefix, suffix):

result = ''

for key, value in pattern.items():

for index, item in enumerate(value):

line = prefix

if index == 0:

line += key

line += delimiter + item + suffix

result += line.strip() + '\n'

print(result)

if __name__ == '__main__':

pattern = {

'エリア': ['日本', 'アメリカ', 'イギリス'],

'性別': ['男', '女'],

'身長': ['100cm以上', '100cm未満']

}

print_pattern: int = None

while True:

print("""

Print dict structure.

Enter print pattern:

1. Product table

2. Category key-value

""")

print_pattern = int(get_input())

if print_pattern in [1, 2]:

break;

delimiter, prefix, suffix = configure()

if print_pattern == 1:

print_product_table(pattern, delimiter, prefix, suffix)

elif print_pattern == 2:

print_category_key_value(pattern, delimiter, prefix, suffix)

The Life

Python: スプレッドシートで各要素の掛け合わせの表を作る

環境

掛け合わせ表の作成 (直積)

補足

分類項目表の作成

まとめたコード

関連記事

この記事を見た人は次の記事も見ています

A Life Summary of an Gypsy