2次元のリストを1次元のリストに展開する方法

Question

シミュレーションに使うパラメータをpythonでヘッダーありのcsvに吐き出すプログラムを作っています．
具体的には

a = [1, 2]
b = [3, 4]

から

a, b
1, 3
1, 4
2, 3
2, 4

というcsvを生成したいのですが，2次元リストを展開して戻り値にするときに文法エラーになってしまいました．
何か良い方法はありますでしょうか．

###発生している問題・エラーメッセージ

SyntaxError: invalid syntax

###該当のソースコード

def csvData(dataSet):
    import itertools
    data = [[] for i in range(len(dataSet))]

    for record in list(itertools.product(*dataSet)):
        for column in range(len(dataSet)):
            data[column].append(record[column])

    return *data

if __name__ == "__main__":
    import sys
    import pandas as pd

    a = [1, 2]
    b = [3, 4]

    a, b = csvData([a, b])

    df = pd.DataFrame({'a':a, 'b':b})
    df.to_csv('param.csv')

Answer 1

itertoolsモジュールのproduct関数を使って、
2つのリストの組み合わせ（直積）を作るのが簡単かと思います。

from itertools import product

a = [1, 2]
b = [3, 4]

print(list(product(a, b)))
# => [(1, 3), (1, 4), (2, 3), (2, 4)]


# CSV
print("a, b")
for x in product(a, b):
    print("%d, %d" % x)

# a, b
# 1, 3
# 1, 4
# 2, 3
# 2, 4

Comments

[hsgwa]

コメントありがとうございます．
考えてみればpandas使う必要なかったんですね．
シンプルな回答ありがとうございました．

Answer 2

参考情報

• pythonでループのネストを減らす定石 http://python.civic-apps.com/reduce-nest-loop/

• tertools.product http://docs.python.jp/3/library/itertools.html#itertools.product

...
product(A, B) は ((x,y) for x in A for y in B) と同じものを返します。
...

Comments

[hsgwa]

参考情報ありがとうございます．お陰様でだいぶスッキリしました．
コメントありがとうございました．

Answer 3

ヘッダーはつけてないので適当にどうぞ

a = [1,2]
b = [3,4]
   
z =  "\n".join([", ".join( ( str(x), str(y) ) ) for x in a for y in b])

print z

Comments

[hsgwa]

コメントありがとうございます
一行だけで解決することだったんですね．
非常に参考になりました．ありがとうございました．