pandasでcsvから取得した情報を条件ごとに結合したいです。

Question

前提・実現したいこと

csvを読み込み値の結合を行いたいです。

今の実行結果だと、
0 ABC111-555-A
1 DEF222-666-
2 GHI333-0-
3 JKL444-0-
となっていますが、条件によって結合のパターンを変えたいと考えています。

付帯番号が0の場合は、番号のみ
記号が存在しない場合は、番号+ '-' +付帯番号
記号が存在する場合は、番号+ '-' +付帯番号+ '-' +記号

上記実行結果を参照すると、
ABC111-555-A
DEF222-666
GHI333
JKL444
という結果になってほしいのですが、if文の条件指定が上手くいきません。

※記号が存在しない場合と記載しましたが、csvの中身を見ると' 'と半角スペースが5つ入っていました。

どなたかご教授お願い致します。
ここに質問の内容を詳しく書いてください。
（例）PHP(CakePHP)で●●なシステムを作っています。
■■な機能を実装中に以下のエラーメッセージが発生しました。

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

エラーメッセージ

該当のソースコード

df = pd.read_csv('infoList.csv', encoding='cp932', na_filter=False)

df['管理番号'] = df['番号'].str.cat(df['付帯番号'].astype(str), sep='-').str.cat(df['記号'].astype(str), sep='-')
url = df['管理番号']
print(url)

試したこと

CSVファイルの中身は下記の通りです。
番号 付帯番号 記号
ABC111 555 A
DEF222 666
GHI333 0
JKL444 0

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

ここにより詳細な情報を記載してください。

Comments

[Daregada]

処理に使うCSVファイルのサンプルを質問文に記載してください。

[m____k]

Daregada様
失礼致しました。
csvの中身は下記の通りです。

番号 付帯番号 記号
ABC111 555 　　A
DEF222 666
GHI333 　　0
JKL444 　　0

[Daregada]

カンマで区切られた元のファイルを提示してください。それとも、元々カンマで区切られていない(CSVではない)のでしょうか?

[m____k]

こちらで！
番号,付帯番号,記号
ABC111,555,A
DEF222,666,
GHI333,0,
JKL444,0,

Answer 1

別解。

import pandas as pd
import io

txt = """
番号,付帯番号,記号
ABC111,555,A
DEF222,666,
GHI333,0,
JKL444,0,
"""

pd.set_option('display.unicode.east_asian_width', True)

df = pd.read_csv(io.StringIO(txt), na_filter=False)

df['管理番号'] = df['番号']
df.loc[df['付帯番号'] != 0, '管理番号'] += "-" + df['付帯番号'].astype(str)
df.loc[df['記号'] != "", '管理番号'] += "-" + df['記号']
print(df['管理番号'])

0    ABC111-555-A
1      DEF222-666
2          GHI333
3          JKL444
Name: 管理番号, dtype: object

Answer 2

こんな感じです。

>>> print(df)
       番号  付帯番号 記号
0  ABC111   555  A
1  DEF222   666
2  GHI333     0
3  JKL444     0
>>>
>>> def mn(row):
...     if row['記号'] != '':
...         return f"{row['番号']}-{row['付帯番号']}-{row['記号']}"
...     elif row['付帯番号'] != 0:
...         return f"{row['番号']}-{row['付帯番号']}"
...     else:
...         return row['番号']
...
>>> df['管理番号'] = df.apply(mn, axis=1)
>>> url = df['管理番号']
>>> print(url)
0    ABC111-555-A
1      DEF222-666
2          GHI333
3          JKL444
Name: 管理番号, dtype: object

Comments

[m____k]

ppaul　様
ありがとうございます！
in で模索していたのですが、rowごとに扱わなければならなかったのですね...
pandasを把握しきれていなかったので勉強になりました！！