質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.48%
Python 3.x

Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

pandas

Pandasは、PythonでRにおけるデータフレームに似た型を持たせることができるライブラリです。 行列計算の負担が大幅に軽減されるため、Rで行っていた集計作業をPythonでも比較的簡単に行えます。 データ構造を変更したりデータ分析したりするときにも便利です。

Q&A

解決済

2回答

3979閲覧

データフレーム中のユニークな組み合わせの頻度を求めたい

Asky

総合スコア17

Python 3.x

Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

pandas

Pandasは、PythonでRにおけるデータフレームに似た型を持たせることができるライブラリです。 行列計算の負担が大幅に軽減されるため、Rで行っていた集計作業をPythonでも比較的簡単に行えます。 データ構造を変更したりデータ分析したりするときにも便利です。

1グッド

0クリップ

投稿2018/08/03 02:44

前提・実現したいこと

tsvを、pandas.DataFrameで読み込み、以下のように二列の要素の組み合わせの頻度を求めて、
組み合わせと頻度のデータフレームを作成したいと考えています。

元のデータ(tsv)

newold
800800
800800
800800
800900
900900
9001000
1000800
1000900
1000900
10001000

作成したいデータフレーム

newoldfreq
8008003
8009001
9009001
90010001
10008001
10009002
100010001

tsvファイルは4列x2,000,000行前後のものが1000ファイルあります。
(一つのtsvファイルに上記の元データの形のデータが横に二つ並んでいるような状態です。)
(1列目と3列目、2列目と4列目の組み合わせの頻度を独立に求めるつもりです。)
要素には200 ~ 1000の数値が入っています。

発生している問題・エラーメッセージ

試しに1ファイルでやってみましたが、読み込みに30秒くらいかかってしまいました。
頻度の列を求めようとすると、五分くらいかかり、tsvファイルが大きいときはブラウザが落ちてしまいます(Jupyter notebookを使っています)。
このままでは1000ファイルを読み込んで一つの頻度列を求められず困っております。

該当のソースコード

Python3

1import os 2import pandas as pd 3 4#ファイル名のリスト作成 5path = "./demo/" 6files = os.listdir(path) 7files_file = [f for f in files if os.path.isfile(os.path.join(path, f))] 8 9#データフレーム初期化 10df1 = pd.DataFrame(index=[], columns=[]) 11df2 = pd.DataFrame(index=[], columns=[]) 12 13# データ読み込み 14def load(df1,df2): 15 for f in files_file: 16 #./demoのtsvを読み込む。ヘッダーは無い。 17 df_tmp = pd.read_table('./demo/'+ f, header = None) 18 #1列目と3列目が対応 19 df_tmp1 = df_tmp.iloc[:,[0,2]] 20 df_tmp1 = df_tmp1.rename(columns={0: 'new', 2: 'old'}) 21 #縦につなげる 22 df1 = pd.concat([df1, df_tmp1]) 23 #2列目と4列目が対応 24 df_tmp2 = df_tmp.iloc[:,[1,3]] 25 df_tmp2 = df_tmp2.rename(columns={1: 'new', 3: 'old'}) 26 #縦につなげる 27 df2 = pd.concat([df2, df_tmp2]) 28 29 print(df.head().append(df.tail())) 30 return df1, df2 31 32df1, df2 = load(df1,df2) 33 34#表示 35df = load()
newold
0864864
1849849
2849849
3849849
4849849
20308038849849
20308039849849
20308040864865
20308041855856
20308042864865

Python3

1# 二つの要素を文字列に変換して結合した列をつくる 2df1['comb'] = df1['new'].astype(str) + df1['old'].astype(str) 3# 結合した文字列の頻度を数えて、vcに入れる 4vc = df1['comb'].value_counts() 5vc

試したこと

上記のように、数値を文字列に変換して結合することで、組み合わせのユニークさを評価できる要素を作りましたが、ここで非常に時間がかかってしまうようです。
数値の組み合わせの頻度をカウントするいい方法はありませんでしょうか?

補足情報(FW/ツールのバージョンなど)

Python 3.6.3
Jupyter notebook 4.4.0

tachikoma👍を押しています

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

回答2

0

groupby().count() で実装すると良いのではないでしょうか。

Python

1res = df.assign(count=0).groupby(['new','old'])['count'].count().reset_index()

とりあえず私の環境では、文字列に変換する方法に対して10倍くらいの速度が得られております

投稿2018/08/03 05:14

magichan

総合スコア15898

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

Asky

2018/08/06 00:31

すぐに実装できて私の環境でも10倍以上早くなりました。ありがとうございます!
guest

0

ベストアンサー

ファイルサイズが大きいとのことなので、一行ずつ処理するのがいいのかもしれません。ファイルの読み込みは標準のcsvモジュール、カウンターとしてはdefaultdictを使うと次のようになります。

1個のファイルへの処理

Python

1import csv 2from collections import defaultdict 3 4counter = defaultdict(int) # intの0で初期化するdict 5filename = "xxx" 6with open(filename) as f: 7 reader = csv.reader(f) 8 for row in reader: 9 key = tuple(row) # listのままではkeyとして使えないのでペアをtupleに変換 10 counter[key] += 1 # 数え上げ

これを複数のファイルに対して繰り返せば所望の結果になるかと思います。
ただしアイディアレベルでパフォーマンスは測ってません。

投稿2018/08/03 04:09

tachikoma

総合スコア3601

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

Asky

2018/08/06 00:31

その後結局ファイル読み込みのところでもpythonが落ちる事態になってしまったので、非常に助かりました。ありがとうございます。
guest

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.48%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問