熊猫数据帧插值而不会丢失数据

时间:2020-05-09 05:25:57

标签: pandas

当熊猫中没有缺失数据时,我该如何插值? 我的原始

enter image description here

我想要的输出:

enter image description here

1 个答案:

答案 0 :(得分:1)

您可以尝试使用索引,然后进行插值:

df.reindex(np.linspace(0, len(df)-1, 2*len(df)-1))\
    .reset_index(drop=True) \
    .apply(lambda x: x.interpolate('linear'))

说明:

主要有两个步骤:每2行插入NaN行,然后进行插值。

  1. 创建一个新列表,将其用作索引。它必须从第一个df索引开始,到最后一个df索引结束,并且值差距为0.5(当前df索引差距的一半)。一个简单的解决方案是使用numpy.linspace
np.linspace(0, len(df)-1, 2*len(df)-1)
  1. 使用reindex进行索引以创建步骤1。

  2. (可选)使用reset_index重置索引。

  3. 使用apply后跟interpolate

    在所有列上应用插值

代码+说明

# import module
import pandas as pd
import numpy as np

# create dataframe
df = pd.DataFrame({'a': [4.18483, 4.15276, 4.13156, 4.1169], 'b': [99.90139, 98.91539, 97.9294, 96.9434]})

# Step 1
print(np.linspace(0, len(df)-1, 2*len(df)-1))
# [0.  0.5 1.  1.5 2.  2.5 3. ]

# Step 2
print(df.reindex(np.linspace(0, len(df)-1, 2*len(df)-1)))
#            a         b
# 0.0  4.18483  99.90139
# 0.5      NaN       NaN
# 1.0  4.15276  98.91539
# 1.5      NaN       NaN
# 2.0  4.13156  97.92940
# 2.5      NaN       NaN
# 3.0  4.11690  96.94340

# Step 3
print(df.reindex(np.linspace(0, len(df)-1, 2*len(df)-1))
      .reset_index(drop=True))
#          a         b
# 0  4.18483  99.90139
# 1      NaN       NaN
# 2  4.15276  98.91539
# 3      NaN       NaN
# 4  4.13156  97.92940
# 5      NaN       NaN
# 6  4.11690  96.94340

# Step 4
print(df.reindex(np.linspace(0, len(df)-1, 2*len(df)-1))
      .reset_index(drop=True)
      .apply(lambda x: x.interpolate('linear')))
#           a          b
# 0  4.184830  99.901390
# 1  4.168795  99.408390
# 2  4.152760  98.915390
# 3  4.142160  98.422395
# 4  4.131560  97.929400
# 5  4.124230  97.436400
# 6  4.116900  96.943400