Python类实例化和返回

时间:2019-07-24 11:46:25

标签: python class dataframe return

我正在尝试通过将它们重写为函数和类来改进代码。该代码旨在从数据框中提取数据并从中创建不同的数据框,然后继续对它们进行更多的数据分析。

当我尝试将它们编写为函数时,它返回元组,而不是我想要的数据框,因此我尝试了类。此处的目标是使代码可以在不同的列中重复使用,在我展示的示例中为“葡萄糖”,但是可以是“乳酸”,“谷氨酸”等。

class ExtractData:
    def GetAnalyte(self, analyte):
        self.analyte = analyte
        reference_data = pd.DataFrame.as_matrix(j[analyte])
        Y_calib = reference_data[:667]
        Y_valid = reference_data[667:]
        return (Y_calib, Y_valid)

    def GetSpectra():
        spec_column = [col for col in j if col.startswith('Signal')]
        spectra_data=pd.DataFrame.as_matrix(j[spec_column])
        X_calib = spectra_data[:667]
        X_valid = spectra_data[667:]
        return(X_calib, X_valid)

gluc = ExtractData()
gluc.GetAnalyte('Glucose')
.

当我在Spyder中运行此代码时,变量资源管理器不会返回任何Y_calib,Y_valid等预期的结果。我错过了什么?任何其他改进都将受到赞赏。

0 个答案:

没有答案