python3でcsvをgzip圧縮する方法

をしようとしてググって出て来る情報がだいたいpython2系だったので、python3で書く方法を記録しておく。

import csv
import gzip
import io
import numpy as np

header = ["foo", "bar", "hoge"]
body = np.random.rand(3, 3)
csv_array = np.vstack([header, body])

stringio = io.StringIO()
writer = csv.writer(stringio)

writer.writerows(csv_array)

data = gzip.compress(stringio.getvalue().encode("utf-8"))
f = open("foo.csv.gz", "wb")
f.write(data)
f.close()

軽くググッた感じだとpython2系だとｃｓｖの書き込みはstringじゃなくてbyteらしいので、もっと簡単にできるんだけど、3系は少し手間がかかる。