新しく「保有期間」列を作りたい。

Question

ここに質問の内容を詳しく書いてください。
画像にある、Dataframeに新しく「保有期間」列を作りたいです。
pythonで株式投資の分析を行っています。
約定履歴のCSVファイルを使って、投資パフォーマンスの分析を行いたいです。
ファイルを読み込んだまでは良かったのですが、新しく「保有期間」の列を作ることができず困っています。

実現したいこと

・「保有期間」の列を追加する
・「保有期間」列の値は、銘柄ごとの約定日の「売付」から「買付」を引いた値

該当のソースコード

import pandas as pd
df = pd.read_csv('tradehistory(JP).csv', encoding='shift-JIS')

試したこと

ファイルを読み込んで、Dataframeに変換する。
使う列が、「約定日」「銘柄名」「売買区分」であることはわかるのですが、
・銘柄名ごとに絞る方法
・銘柄ごとにソートして、売付ー買付をする方法
が、わかりません。

どなたかお分かりの方がいらっしゃいましたら、お手本のコードをご教示いただきたいです。
よろしくお願いします。

Answer 1

※ 銘柄ごとに買付・売付が一組だけある、という前提

import pandas as pd

df = pd.read_csv('tradehistory(JP).csv', encoding='shift-JIS', parse_dates=['約定日'])
df['保有期間'] = df.groupby('銘柄名')\
                   .apply(lambda x: x['約定日'].sort_values().diff())\
                   .droplevel(0).fillna(pd.Timedelta(seconds=0))
print(df)

#
      約定日            銘柄名 売買区分 保有期間
0 2019-03-20      アールビバン     買付   0 days
1 2019-03-25  タキロンシーアイ     買付   0 days
2 2019-03-27      アールビバン     売付   7 days
3 2019-03-28  タキロンシーアイ     売付   3 days
4 2019-03-29            テモナ     買付   0 days

Comments

[papermount]

ご回答いただき、ありがとうございます。
欲しい結果を得ることができました！
ありがとうございます。