Pythonでitertoolsモジュールを使って直積を求める方法 【初心者向け】

初心者向けにPythonで直積を求める方法について解説しています。直積を求めるにはitertoolsモジュールを使用します。基本構文と実行結果をサンプルで確認しましょう。

TechAcademyマガジンはオンラインのプログラミングスクールTechAcademy [テックアカデミー]が運営する教育×テクノロジーのWebメディアです。初心者でもすぐ勉強できる記事が2,000以上あります。

Pythonで直積を求める方法 (itertoolsモジュールの使い方)について解説します。

Pythonについてそもそもよく分からないという方は、Pythonとは何なのか解説した記事をまずご覧ください。

 

なお本記事は、TechAcademyのPythonオンライン講座の内容をもとにしています。

 

田島悠介

今回は、Pythonに関する内容だね!

大石ゆかり

どういう内容でしょうか?

田島悠介

直積を求める方法について詳しく説明していくね!

大石ゆかり

お願いします!

 

直積とは

直積とは、複数の集合から要素を一つずつ取り出して組にしたものです。直積集合やデカルト積とも呼ばれます。

Wikipedia

例えば上記の場合、A={x, y, z}、B={1, 2, 3}なので、直積A×Bは以下のようになります。

A×B={(x,1),(x,2),(x,3),(y,1),(y,2),(y,3),(z,1),(z,2),(z,3)}

 

直積を求める方法(itertools.productの使い方)

Python で直積を求めるには、 itertools.product を使います。 itertools は直積の他、順序や組み合わせを簡潔に書くことができる標準ライブラリです。直積を求める書き方は以下のとおりです。

import itertools

itertools.product(集合1, 集合2, ...)

戻り値は itertools.product 型となります。以下のように list の引数に指定することで list 型に変換できます。

list(itertools.product(集合1, 集合2, ...))

集合は複数指定可能です。集合を指定しないと空の itertools.product 型が返ります。また集合を1つだけ指定すると、もとの集合と要素数が同じ組み合わせが返ります。

 

[PR] Pythonで挫折しない学習方法を動画で公開中

実際に書いてみよう

今回のサンプルプログラムでは、 itertools.product を使った直積の求め方を確認します。また itertools.product を使わない場合と比較し、どのぐらい簡潔に書くことができるかを確認します。

はじめに、itertools.product を使った直積の求め方です。見やすくするため list に変換しています。

import itertools

A = ('x', 'y', 'z')
B = (1, 2, 3)

list(itertools.product(A, B))

実行結果は以下のようになります。

[('x', 1),
 ('x', 2),
 ('x', 3),
 ('y', 1),
 ('y', 2),
 ('y', 3),
 ('z', 1),
 ('z', 2),
 ('z', 3)]

次に、itertools.product を使わずに直積を求めてみましょう。

AB = []
for i in A:
    for j in B:
        AB.append((i, j))
print(AB)

実行結果は以下のようになります。

[('x', 1),
 ('x', 2),
 ('x', 3),
 ('y', 1),
 ('y', 2),
 ('y', 3),
 ('z', 1),
 ('z', 2),
 ('z', 3)]

itertools.product を使うと、 for 文をネストさせずに直積を求めることができます。

 

この記事を監修してくれた方

太田和樹(おおたかずき)
ITベンチャー企業のPM兼エンジニア

普段は主に、Web系アプリケーション開発のプロジェクトマネージャーとプログラミング講師を行っている。守備範囲はフロントエンド、モバイル、サーバサイド、データサイエンティストと幅広い。その幅広い知見を生かして、複数の領域を組み合わせた新しい提案をするのが得意。

開発実績:画像認識技術を活用した駐車場混雑状況把握(実証実験)、音声認識を活用したヘルプデスク支援システム、Pepperを遠隔操作するアプリの開発、大規模基幹系システムの開発・導入マネジメント

地方在住。仕事のほとんどをリモートオフィスで行う。通勤で消耗する代わりに趣味のDIYや家庭菜園、家族との時間を楽しんでいる。

 

大石ゆかり

内容分かりやすくて良かったです!

田島悠介

ゆかりちゃんも分からないことがあったら質問してね!

大石ゆかり

分かりました。ありがとうございます!

オンラインのプログラミングスクールTechAcademyではPythonを使って機械学習の基礎を学ぶPythonオンライン講座を開催しています。

初心者向けの書籍を使って人工知能(AI)や機械学習について学ぶことができます。

現役エンジニアがパーソナルメンターとして受講生に1人ずつつき、マンツーマンのメンタリングで学習をサポートし、最短4週間で習得することが可能です。

また、現役エンジニアから学べる無料のプログラミング体験会も実施しているので、ぜひ参加してみてください。