[pandas]pythonのpandasデータフレームで動的に列追加するコード

Question

あいうえお
かきくけこ
さしすせそ
たちつてと
なにぬねの
はひふへほ
まみむめも
やゆよ
らりるれろ
わをん
あかさたな
はまやらわ
いきしちに
ひみり
うくすつぬ
ふむゆる

こんな感じのファイルがあったとして、これを、
1行を1つの単語として扱い、そして、5行ごとを1つの文書として扱いSCDVしたいのですが、その前準備として、このファイルを

0 あいうえお
0 かきくけこ
0 さしすせそ
0 たちつてと
0 なにぬねの
1 はひふへほ
1 まみむめも
1 やゆよ
1 らりるれろ
1 わをん
2 あかさたな
2 はまやらわ
2 いきしちに
2 ひみり
2 うくすつぬ
3 ふむゆる

のようなデータフレームにする必要がありそうなのですが、
pandasデータフレームにこのように文書番号の列を追加するコードがわかりません。

# coding: UTF-8

#必要なライブラリをインポート
import pandas as pd

#

df = pd.read_csv('tekitou.txt',sep=('\n'),header=None)#ファイル読み込み

df['text_num'] = 0#とりあえずハリボテの列を追加

この後どのようなコードを書けば考えている通りの列を追加できるでしょうか。
お手数ですがご教示よろしくお願いします。

Answer 1

2例提示します。

l = len(df.index)
num = int(l/5) + 1
print("num :", num)
no = [y for y in range(num) for x in [0]*5]
print(no)

no = no[:l]
df.loc[:,'text_num'] = no
print(df)

no = df.reset_index().loc[:,"index"].apply(lambda x: x // 5)
df.loc[:,'text_num'] = no
print(no)
print(df)

Comments

[Terratale]

回答ありがとうございます。
先にズラーっと文書番号を並べた後でそれを入れるという方法があったんですね！
無事に狙い通りのデータフレームを作成することができました！

Answer 2

取り急ぎ、列追加のコードだけでも。
pandas.DataFrameに列や行を追加

df['A'] = 0

Comments

[Terratale]

回答ありがとうございます！
参考になるページありがとうございました。