PythonのDF内で使用する関数について

Question

Python初心者でどのような関数、メソッドを使用すればよいかわからず質問させて頂きます。

PythonのpandasでエクセルファイルをDFとして取り込んだ後、下記のような処理をDF内で実施したいです。

上記のxlsxファイルをpandasで取り込んだ後、
①C列の“quantity”とE列の“price”を乗じ、新しく列I“total amount”にその解を表示させたい（下図I列ご参照）
→リストの考え方がよく分かっていませんでしたが、リスどうしで加減乗除することが理解できました。
②このファイル名は、“df_NY_test data.xlsx”であるが、ファイル名の一部分“NY”をB列に挿入したい（下図B列ご参照）

→正規表現によるマッチングについて理解を深めることができました。また、文字集合についても理解度が足りなかったため、再度復習しました。

関数、メソッド等まだ正しく理解が進んでおらず、何を書けばよいのか途方に暮れています。
誠にお手数をおかけしますが、アドバイスを頂ければ大変ありがたいです。
何卒宜しくお願い致します。

Comments

[Zhiye]

丸投げの質問をしてしまい申し訳ありませんでした。
re モジュール、re.sub()、また、文字集合について理解が全く足りていなかったため、頂いたコードを見て再度復習します。
また、リストについてこちらも理解が足りていないことを痛感しました。
ご指摘を真摯に受け止め、次回は一度コードを書いた上で質問いたします。

Answer 1

import pandas as pd
import re

target = 'df_NY_test data.xlsx'
location = re.sub(r"^.*?df_(.+?)_.*$", r"\1", target)

df = pd.read_excel(target, index_col=0)
df['total amount'] = df['quantity'] * df['price']
df.insert(0, 'location', location)
print(df)

  location name  quantity unit  price Ounit  price unit  total amount
1       NY    a        10   MM     10    MM           1           100
2       NY    b        15   ST     20    ST          10           300
3       NY    c       200   MM     30    MM         100          6000
4       NY    d        25   MM     15    MM           1           375
5       NY    e        30   ST     40    ST          10          1200

Comments

[Zhiye]

早々にありがとうございました。また、質問が丸投げになってしまい、本当に申し訳ありません。
頂いたご回答をしっかりと学び、自分で書けるようにいたします。
重ねてにはなりますが、ありがとうございました。