データフレームで数字の繰り返しを行いたい

Question

お世話になります。
データフレームで数字の繰り返し方法についてご教示願います。
以下のデータフレームでBには1と-1を繰り返す処理を、Cには123を繰り返す処理をしたいと思っております。

A	B	C
1	1	1
2	-1	2
3	1	3
4	-1	1
5	1	2
6	-1	3
：	：	：

Bについては以下のようにやってみましたが、print文の表示では上手くいくものの、なぜかCSVに落とすと欠損のままです。
綺麗ではないので、もっとスマートな方法はないものでしょうか。

Cの方法は思いつきません・・・。

A=np.arange(1,200)
df=pd.DataFrame({"A":n})
df["B"]=1
df["B"]=df["B"][0::2]
df.fillna(-1)
print(df.fillna(-1))

よろしくお願いいたします。

Answer 1

こんな感じでしょうか

import numpy as np
import pandas as pd
A=np.arange(1,200)
df=pd.DataFrame({"A":A})

def f(r):
    r['B'] = -1 if r['A'] % 2 == 0 else 1
    r['C'] = (r['A']-1) % 3 + 1
    return r

df = df.apply(f,axis=1)
print(df)
"""
       A  B  C
0      1  1  1
1      2 -1  2
2      3  1  3
3      4 -1  1
4      5  1  2
5      6 -1  3
"""

Comments

[sham0909]

思い通りの結果になりました。
ありがとうございました。