Pythonのリスト内包表記を用いてtxtファイルから有向グラフを作成

Question

###前提・実現したいこと

HOGE1 HOGE2
HOGE3 HOGEHOGE
HOGE5 HOGE10

ファイルに、2点を結ぶ辺を上記のように記述してある(約10000行)ので、これから
[(HOGE1,HOGE2),(HOGE3,HOGEHOGE),(HOGE5,HOGE10)]というリストを作りたいです。

最終的に実現したいことは、ファイルから有向グラフを作成することなので、もしこのやり方よりも簡単なやり方があれば教えて頂きたいです。

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

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import networkx as nx
import matplotlib.pyplot as plt

with open('sample.txt', "r") as f:
    data = [v.replace(" ", ",").rstrip('\r\n') for v in f.readlines()]

G = nx.DiGraph(test)
nx.draw_networkx(G)
plt.show()

###試したこと
['HOGE1,HOGE2','HOGE3,HOGEHOGE','HOGE5,HOGE10']
とするところまでは考えられたのですが、これをタプルにする方法がわかりません。

###補足情報(言語/FW/ツール等のバージョンなど)
Python3.6

Answer 1

こんな感じですか？

from pprint import pprint

with open("sample.txt") as f:
    r = [tuple(_ for _ in s.rstrip().split(" ")) for s in f.readlines()]
pprint(r)
"""
[('HOGE1', 'HOGE2'), ('HOGE3', 'HOGEHOGE'), ('HOGE5', 'HOGE10')]
"""

Comments

[退会済みユーザー]

ありがとうございます。
無事有向グラフを作成できました。

Answer 2

from io import StringIO

fp = StringIO('''\
HOGE1 HOGE2
HOGE3 HOGEHOGE
HOGE5 HOGE10
''')

data = (s.split() for s in fp)
print(*data, sep='\n')