リストから要素を取り出して隣に並べたときに表示されるNaNを消したいです

Question

実現したいこと

こちらの関連質問です。

test.csvは以下のようなファイルです。

x
0
1
2
3

今、x列の最初の2行だけを取り出し、x列の横に加えたいです。
つまり、欲しい結果は以下のようになります。

x	x02
0	0
1	1
2	
3	

「該当のソースコード」を実行すると、

   x  x02
0  0  0.0
1  1  1.0
2  2  NaN
3  3  NaN

になってしまいます。
このNaNを消したいです。

該当のソースコード

import pandas as pd

df = pd.read_csv('test.csv')

df['x02'] = df['x'][0:2]

print(df)
#   x  x02
#0  0  0.0
#1  1  1.0
#2  2  NaN
#3  3  NaN

試したこと

「要素数を2個減らす」「NaNを削除」で検索しましたが、今回のような場合に当てはまるものを見つけられませんでした。

「該当のソースコード」の下に、以下のようなコードを加えてみましたが、うまくいきませんでした。

df['x03'] = df['x02'].dropna()
print(df)
#   x  x02  x03
#0  0  0.0  0.0
#1  1  1.0  1.0
#2  2  NaN  NaN
#3  3  NaN  NaN

Answer 1

pandas.DataFrame.fillna()メソッドでNaNを空文字列に変換することで実現はできますが、x02列のデータ型がobject(文字列)になってしまいます。そのため、数値として扱う(演算する)ことができなくなります。

import pandas as pd

df = pd.read_csv('test.csv')

df['x02'] = df['x'][0:2]
df['x02'] = df['x02'].fillna('')

print(df)
df.info()

#    x  x02
# 0  0  0.0
# 1  1  1.0
# 2  2     
# 3  3     
#
# <class 'pandas.DataFrame'>
# RangeIndex: 4 entries, 0 to 3
# Data columns (total 2 columns):
#  #   Column  Non-Null Count  Dtype 
# ---  ------  --------------  ----- 
#  0   x       4 non-null      int64 
#  1   x02     4 non-null      object
# dtypes: int64(1), object(1)
# memory usage: 196.0+ bytes

Comments

[yyicp]

できました。
どうもありがとうございました。