python - csv ファイル内の列のデータを 2 つの個別の出力 csv ファイルに分割する方法は?

Question

私は .csv ファイルを持っています。

ID NAME CATEGORIES
1, x,   AB
2, xx,  AA
3, xxx, BA

カテゴリに基づいて 2 つの出力 .csv ファイルを作成するには、どうすればよいでしょうか。

ファイル 1:

ID NAME CATEGORY
1, x,   A
2, xx,  A
3, xxx, B

ファイル 2:

ID NAME CATEGORY
1, x,   B
2, xx,  A
3, xxx, A

入力と出力をセットアップしましたが、困惑した空の for ループだけです。

records = [line for line in csv.reader(open('test_input.csv', 'rt'), delimiter=',')]
outfile = open('test_output1.csv', 'wt')
outfileWriter = csv.writer(outfile, delimiter=',')

for record in records:
    #something!
    outfileWriter.writerow(record)

outfile.close()

助けていただければ幸いです！