PythonでCSVファイルのデータをset型で読み込む方法

Question

現在CSVファイルからデータを読み込む方法で困っています。

route_key={}
with open('/Users/01.csv', 'r') as f:
    reader = csv.DictReader(f, delimiter=' ')
    for row in reader:
        route_key[row['Rn']] = row['Customer']
        
        
print(route_key)

現在このCSVファイルのデータをPython上で使いたいと考えているのですが、Typeをstrからset型に変更する方法がわかりません。
int型に変えることはできたのですが、そこからどうすればいいのかわかりません。

詳しい方、ご教授のほどよろしくお願いします。

Comments

[Daregada]

読み込むCSVファイルがの仕様が明確でない(空白区切りとカンマ区切りの混合?)ので、質問文に実際のCSVファイル(長ければ先頭の数行)をテキストとして載せて下さい。

[KGO12]

コメントいただきありがとうございます。
現在外出中のため、夕方以降にテキストを追加します。

空白区切りとカンマ区切りの混合で作成しています。

[Daregada]

なるほど。

Rn Customer
R1 1,3
R2 1,2
R3 2,4
R4 3
R5 6
R6 1,2,3,4,5,6

こんな感じで合ってます?

[meg_]

同じコードが2回掲載されています。見にくいので1つにしてもらえませんか？

[KGO12]

Daregada様
その内容で合っています。

[KGO12]

meg_様
編集いたしました。

Answer 1

空白区切りとカンマ区切りの混合で作成しています。

とのことなので、対象となるファイル/Users/01.csvが以下の形式だと仮定してコードを組む。

Rn Customer
R1 1,3
R2 1,2
R3 2,4
R4 3
R5 6
R6 1,2,3,4,5,6

各行のRnを辞書のキー、Customerに書かれたカンマ区切りの数値をintにした集合(set)を値として使えるようにするには、

import csv
from pprint import pprint

route_key = {}
with open('/Users/01.csv', 'r') as f:
    reader = csv.DictReader(f, delimiter=' ')
    for row in reader:
        route_key[row['Rn']] = set([int(x) for x in row['Customer'].split(',')])


pprint(route_key)

{'R1': {1, 3},
 'R2': {1, 2},
 'R3': {2, 4},
 'R4': {3},
 'R5': {6},
 'R6': {1, 2, 3, 4, 5, 6}}

Comments

[KGO12]

早速のご回答ありがとうございます。
後ほど試させていただきます。

[KGO12]

連絡が遅くなり申し訳ありません。
route_key[row['Rn']] = set([int(x) for x in row['Customer'].splint(',')])

KeyError: 'Customer'

このようにKeyErrorとなってしまいます。何が問題なのでしょうか。
Macで編集しており、EncodingはUnicode（UTF-8)です。

[Daregada]

読み込むファイルの一行目に「Customer」と書いてない(スペルミスなど)のでは。
あるいは「Rn」との間が空白一つではない(カンマとかタブとか)可能性も。

[KGO12]

仰るとおりタブで区切っていました。
タブとカンマの場合では先ほどのコードではできないのでしょうか？

delimiter='\t'を入れようと考えています

[Daregada]

ご自分で「reader = csv.DictReader(f, delimiter=' ')」と書いているので、空白区切りで読み込もうとします。実際のファイルに合わせてdelimiterを設定すればいいだけです。

[Daregada]

「空白区切りとカンマ区切りの混合で作成しています。」とあなたが書き、要望を出しても元のファイルが公開されない状況で、その言を信用して空白区切りで動くコードを書いたわけです。

そうしたら、「実際のファイルの内容がタブ区切りなのでエラーが出ます」と言われても困りますがな。それはあなたが自分で直してください。

[KGO12]

ありがとうございます。無事解決することができました。
また、失礼な要望をしてしまい申し訳ありませんでした。