dataset.py 文件源码

def _resample_rdp(self, n):
        """
        Resample Radial Density Profile
        :param n: int
        :return: nothing
        """
        columns = []
        for i in xrange(n):
            columns += ['edp_{}'.format(i)]
        self.feature_names += columns

        # edp = [None] * len(self._df)
        edp = np.zeros([len(self._df), n])
        for i, v in enumerate(self._df.RadialDensityProfile.values):
            x = np.zeros([len(v)])
            for j, v_ in enumerate(v):
                x[j] = v_[0]
            edp[i] = signal.resample(x, n)
        df = pd.DataFrame(data=edp, columns=columns, index=self._df.index)
        self._df = pd.concat([self._df, df], axis=1)