python - 如何将一列数据框或单行数据框作为数据框或系列返回？

Question

给df，

df = pd.DataFrame({'col1':np.arange(6), 'col2':[*'abcdef']})

   col1 col2
0     0    a
1     1    b
2     2    c
3     3    d
4     4    e
5     5    f

然后在选择单个列时，使用:

df['col1']
# returns a pd.Series

0    0
1    1
2    2
3    3
4    4
5    5
Name: col1, dtype: int32

同样选择单行时，

df.loc[0]
# returns a pd.Series

col1    0
col2    a
Name: 0, dtype: object

我们如何强制单列或单行选择返回 pd.DataFrame？

Answer 1

你也可以使用to_frame:

# One column
>>> df['col1'].to_frame()
   col1
0     0
1     1
2     2
3     3
4     4
5     5

# One row
>>> df.loc[0].to_frame().T
  col1 col2
0    0    a