pythonを用いたリツイート関係のリスト化

Pythonを使ってリツイートの関係性をリスト化したいと考えています。
pandasを使って以下のようなCSVファイルを読み込み、リストを作成します。
csvファイルは下のような内容になっています。

tweet_id|text |user_id|date|
|:--|:--:|--:|
|11108....|アイウエオ|daikonman|19:00:00
|11244...|RT @daikonman:アイウエオ|ninjinman|19:30:00
|1167...|RT @daikonman:アイウエオ|renkonman|19:35:00

このようなCSVファイルから、

	from	to
0	daikonman	ninjinman
1	daikonman	renkonman

といった内容のRT関係を表したリストを作成したいと考えているのですが、手順が見つかりません。
何かアドバイスをいただけたら幸いです。

mather

2021/04/28 15:21

手順が見つかりません、ではなく、ご自身でどうやればいいのか模索してください。具体的に何に困っているのでしょうか？少なくとも pandas を使ったコードは書いていると思いますので、そちらを提示しながらどの部分が解決できないのか示しましょう。

Daikonman

2021/05/02 08:39

いただいたご意見の通り、自分で模索した結果なんとか解決できました。ありがとうございました。

行動規範の内容に同意します

回答1件

自己解決

１行ずつfor文で回すことで解決できました。
コードをいかに載せておきます。
おそらく大規模なデータでは時間がかかる処理になっています。

python
1import pandas as pd
2
3df = pd.read_csv('Example.csv', parse_dates=['date'])
4rt_from = []
5rt_to = []
6
7for i, j in df.iterrows(): 
8  t = str(j['text'])
9  if t[0:2] == 'RT':
10    target1 = '@'
11    target2 = ':'
12    idx1 = t.find(target1)
13    idx2 = t.find(target2)
14    rt_from.append(t[idx1+1:idx2]) 
15    rt_to.append(j['user_id'])
16  else:
17    continue
18
19
20df_rt_relation = pd.DataFrame({'from': rt_from, 'to': rt_to})
21
22df_rt_relation.to_csv("rt_relation.csv")