pythonでデータフレームを編集し特定の列の数字毎に表を作りたい。

Question

データフレームのCD毎にレコードを表記したいです。
今後CDが増えることを見据えfor分でCDの数字を見に行き、数字が一致していれば表を作成するようにしたいです。個人的には変数iとCDの数字が一致という条件の部分ががうまくいっていないと思っています。よろしくお願いします。

import pandas as pd
import numpy as np

df = pd.DataFrame({
'city': ['osaka', 'osaka', 'osaka', 'osaka', 'tokyo', 'tokyo', 'tokyo'],
'food': ['apple', 'orange', 'banana', 'banana', 'apple', 'apple', 'banana'],
'price': [100, 200, 250, 300, 150, 200, 400],
'CD': [20, 11, 20, 4, 20, 11, 11]})
df
for i in range (100):
if i == ("CD == i"):
df1 = df.query("CD == i")
print(df1)

作成後イメージ
CD4
|city|food|price|CD
|:--|:--:|--:|
|osaka|banana|300|4

CD11
|city|food|price|CD
|:--|:--:|--:|
|osaka|oraneg|200|11
|tokyo|apple|200|11
|tokyo|banana|400|11

CD20
|city|food|price|CD
|:--|:--:|--:|
|osaka|apple|100|20
|osaka|banana|250|20
|tokyo|apple|150|20

Answer 1

以下のようにCD列の値を抽出してループ処理すればよいです。

import pandas as pd

df = pd.DataFrame({
'city': ['osaka', 'osaka', 'osaka', 'osaka', 'tokyo', 'tokyo', 'tokyo'],
'food': ['apple', 'orange', 'banana', 'banana', 'apple', 'apple', 'banana'],
'price': [100, 200, 250, 300, 150, 200, 400],
'CD': [20, 11, 20, 4, 20, 11, 11]})

cds = sorted(set(df['CD']))
for cd in cds:
    df1 = df.query(f'CD == {cd}')
    print(df1)

なお、一部の列名にてqueryが正確に動作しないようです。
たとえばコード２桁がコード2桁のように数字部分が半角として解釈されてしまうようです。
よってqueryを使わない以下のようなコードのほうが確実です。

import pandas as pd

df = pd.DataFrame({
'city': ['osaka', 'osaka', 'osaka', 'osaka', 'tokyo', 'tokyo', 'tokyo'],
'food': ['apple', 'orange', 'banana', 'banana', 'apple', 'apple', 'banana'],
'price': [100, 200, 250, 300, 150, 200, 400],
'コード２桁': [20, 11, 20, 4, 20, 11, 11]})

cds = sorted(set(df['コード２桁']))
print(cds)
for cd in cds:
    # pandas.core.computation.ops.UndefinedVariableError: name 'コード2桁' is not defined
    #df1 = df.query(f'コード２桁 == {cd}')
    #df1 = df.query(f'`コード２桁` == {cd}')

    # OK
    df1 = df[df['コード２桁'] == cd]
    print(df1)

Comments

[fire]

コメントありがとうございます。
非常にわかりやすかったのですが11行目の(f'CD == {cd}')のfと{}は何を意味するのでしょうか？
また少し列名を変えて列名の「CD」を「CD(2桁)」や「コード２桁」に変えるとエラーが出ます。
大文字小文字や数字や記号が混じるとエラーになるのでしょうか？勉強の為教えて頂ければありがたいです。
よろしくお願いいたします。

[8524ba23]

fについては「fstring python」で検索すると情報が得られます。
列名については、どのようにコードを変更したのか不明ですが、数字や記号が入っても問題はないです。

[fire]

何度もすみません。変更したもの２つ添付しました。
よろしくお願いします。
・コード２桁に変更したコード
import pandas as pd
df = pd.DataFrame({
'city': ['osaka', 'osaka', 'osaka', 'osaka', 'tokyo', 'tokyo', 'tokyo'],
'food': ['apple', 'orange', 'banana', 'banana', 'apple', 'apple', 'banana'],
'price': [100, 200, 250, 300, 150, 200, 400],
''コード２桁: [20, 11, 20, 4, 20, 11, 11]})

cds = sorted(set(df['コード２桁']))
for cd in cds:
df1 = df.query更したコード。(f'コード２桁 == {cd}')
print(df1)

・CD(2桁)に変更したコード
import pandas as pd

df = pd.DataFrame({
'city': ['osaka', 'osaka', 'osaka', 'osaka', 'tokyo', 'tokyo', 'tokyo'],
'food': ['apple', 'orange', 'banana', 'banana', 'apple', 'apple', 'banana'],
'price': [100, 200, 250, 300, 150, 200, 400],
'CD(2桁)': [20, 11, 20, 4, 20, 11, 11]})

cds = sorted(set(df['CD(2桁)']))
for cd in cds:
df1 = df.query(f'CD(2桁) == {cd}')
print(df1)

[8524ba23]

列名「コード２桁」にてエラー再現したため、queryを使わない方法を追記しました。

[fire]

解決しました。ありがとうございました。