python pandasのdf.loc[]へ代入するとエラーが出る

Question

やりたいこと

PythonでCSVを読み込み、読み込んだCSVの行をfor文で回し、
indexの次の番号の変数を作って（現在indexが１なら次のindex 2）とカラム名を指定して代入したいので、df.loc[変数,列名]で出来ないかと思い、実行したところ、次のようなエラーが出ました。（特定の列を次の列のものにずらしたい）。このコードだとindexが５（無いindex）まで指定してしまうのでエラーが出るのかと思い修正しましたが、エラーは消えませんでした。これは一体なんなのか教えていただけたら幸いです。

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

raise KeyError(key) from err
KeyError: 4

該当のソースコード

import pandas as pd
import csv

df = pd.read_csv('test.csv',encoding='utf_8')
data = []
index_next = 0

for index,row in df.iterrows():#1行ずつ呼び出す
    print("index:",index)
    index_next = index+1#indexに1を足したもの

    if index_next < len(df.index):
        print("index_next:",index_next)
        print("row:",row)

        index_column = df.loc[index_next, "gram"]#今処理している行の次の行のカラム"gram"を抽出したい
        print(index_column)

name,gram,test
Akiko,300g,1
Ken,200g,2
Taro,500g,3
Keiko,800g,4

試したこと

index_nextでなく、indexをそのまま入れた場合
index_column = df.loc[index, "gram"]や、
数値をそのまま入れた場合
index_column = df.loc[2, "gram"]はエラーは出ませんでした。
またtypeで型を出すと、indexもindex_nextもint型でした。

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

ライブラリ：pandas,csv
環境：Python3.8.9 MacOS 12.3.1、ターミナルで実行

pythonやpandasを使い始めて間もないため色々理解しきれていませんが、アドバイスいただけると助かります。

Comments

[melian]

データフレームは4行なので、インデックスは 0 から 3 になります。

[melian]

index_next = index + 1 の位置が違うので、以下の様にします。

for index,row in df.iterrows():#1行ずつ呼び出す
　print("index:",index)
　index_next = index+1#indexに1を足したもの
　if index_next < len(df.index):
　　print("index_next:",index_next)

[oyatsu8]

ありがとうございます。修正してみます。
回答でないのでmelianさんの評価ができないのですが、お忙しいと思いますが、回答側に書いてもらえないでしょうか

[melian]

回答へ転記しました。oyatsu8 さんが期待する最終的な結果(データフレームかリストでしょうか)は具体的にはどの様なものなのでしょう？

[oyatsu8]

ありがとうございます。最終的にはjsonで書き出したいと思っています。このような形にしたいです。

[{"name":"Akiko","gram":"300g","test":2},
{"name":"Akiko","gram":"200g","test":3},
{"name":"Akiko","gram":"50g","test":4},
{"name":"Akiko","gram":"800g","test":none}]

[melian]

回答に追記しました。参考にしてみて下さい。

[jbpb0]

> うまくいきました。

質問を「解決済」にしてください

Answer 1

※ コメント欄から転記

index_next = index + 1 の位置が違いますので、以下の様に変更します。

for index,row in df.iterrows():#1行ずつ呼び出す
    print("index:",index)
    index_next = index+1#indexに1を足したもの
    if index_next < len(df.index):
        print("index_next:",index_next)

追記

最終的にはjsonで書き出したいと思っています。このような形にしたいです。

JSON への書き出しには pandas.DataFrame.to_json — pandas 1.5.3 documentation というメソッドを利用することができます。ただ、ドキュメントにも書かれていますが、NaN や None は null に置き換えられます。

import pandas as pd

df = pd.read_csv('test.csv',encoding='utf_8')

# test 列を上書き
df['test'] = df.index + 2
# test 列の最終行に None(NaN) を入れる
df.loc[len(df)-1, 'test'] = None
# データフレームをJSON フォーマットな文字列に変換
json_data = df.to_json(orient='records', double_precision=0)

print(json_data)

# [{"name":"Akiko","gram":"300g","test":2},
#  {"name":"Ken","gram":"200g","test":3},
#  {"name":"Taro","gram":"500g","test":4},
#  {"name":"Keiko","gram":"800g","test":null}]

Comments

[oyatsu8]

melian様、ありがとうございます。うまくいきました。
ループ処理をするときにいつも頭がこんがらがってしまい、、助かりました。

[oyatsu8]

JSON への書き出しまで教えてくださりありがとうございます。このような書き方ができるんですね、とても参考になりました。