पंडों DataFrame के उपवर्ग के लिए संपत्ति सेटर

Question

मैं कोशिश कर रहा हूँ pd.DataFrameकि जब ( groupऔर timestamp_col) को इनिशियलाइज़ किया जाए तो उसमें दो आवश्यक तर्क हो । मैं उन तर्कों पर सत्यापन चलाना चाहता हूं groupऔर timestamp_colइसलिए, मेरे पास प्रत्येक गुण के लिए एक सेटर विधि है। यह सब तब तक काम करता है जब तक मैं कोशिश करता हूं set_index()और प्राप्त करता हूं TypeError: 'NoneType' object is not iterable। ऐसा प्रतीत होता है कोई बहस में मेरी सेटर कार्य करने के लिए पारित किया जा रहा है test_set_indexऔर test_assignment_with_indexed_obj। यदि मैं if g == None: returnअपने सेटर फ़ंक्शन में जोड़ता हूं, तो मैं परीक्षण मामलों को पास कर सकता हूं लेकिन यह मत सोचिए कि यह उचित समाधान है।

मुझे इन आवश्यक तर्कों के लिए संपत्ति सत्यापन कैसे लागू करना चाहिए?

नीचे मेरी कक्षा है:

import pandas as pd
import numpy as np


class HistDollarGains(pd.DataFrame):
    @property
    def _constructor(self):
        return HistDollarGains._internal_ctor

    _metadata = ["group", "timestamp_col", "_group", "_timestamp_col"]

    @classmethod
    def _internal_ctor(cls, *args, **kwargs):
        kwargs["group"] = None
        kwargs["timestamp_col"] = None
        return cls(*args, **kwargs)

    def __init__(
        self,
        data,
        group,
        timestamp_col,
        index=None,
        columns=None,
        dtype=None,
        copy=True,
    ):
        super(HistDollarGains, self).__init__(
            data=data, index=index, columns=columns, dtype=dtype, copy=copy
        )

        self.group = group
        self.timestamp_col = timestamp_col

    @property
    def group(self):
        return self._group

    @group.setter
    def group(self, g):
        if g == None:
            return

        if isinstance(g, str):
            group_list = [g]
        else:
            group_list = g

        if not set(group_list).issubset(self.columns):
            raise ValueError("Data does not contain " + '[' + ', '.join(group_list) + ']')
        self._group = group_list

    @property
    def timestamp_col(self):
        return self._timestamp_col

    @timestamp_col.setter
    def timestamp_col(self, t):
        if t == None:
            return
        if not t in self.columns:
            raise ValueError("Data does not contain " + '[' + t + ']')
        self._timestamp_col = t

यहाँ मेरे परीक्षण के मामले हैं:

import pytest

import pandas as pd
import numpy as np

from myclass import *


@pytest.fixture(scope="module")
def sample():
    samp = pd.DataFrame(
        [
            {"timestamp": "2020-01-01", "group": "a", "dollar_gains": 100},
            {"timestamp": "2020-01-01", "group": "b", "dollar_gains": 100},
            {"timestamp": "2020-01-01", "group": "c", "dollar_gains": 110},
            {"timestamp": "2020-01-01", "group": "a", "dollar_gains": 110},
            {"timestamp": "2020-01-01", "group": "b", "dollar_gains": 90},
            {"timestamp": "2020-01-01", "group": "d", "dollar_gains": 100},
        ]
    )

    return samp

@pytest.fixture(scope="module")
def sample_obj(sample):
    return HistDollarGains(sample, "group", "timestamp")

def test_constructor_without_args(sample):
    with pytest.raises(TypeError):
        HistDollarGains(sample)


def test_constructor_with_string_group(sample):
    hist_dg = HistDollarGains(sample, "group", "timestamp")
    assert hist_dg.group == ["group"]
    assert hist_dg.timestamp_col == "timestamp"


def test_constructor_with_list_group(sample):
    hist_dg = HistDollarGains(sample, ["group", "timestamp"], "timestamp")

def test_constructor_with_invalid_group(sample):
    with pytest.raises(ValueError):
        HistDollarGains(sample, "invalid_group", np.random.choice(sample.columns))

def test_constructor_with_invalid_timestamp(sample):
    with pytest.raises(ValueError):
        HistDollarGains(sample, np.random.choice(sample.columns), "invalid_timestamp")

def test_assignment_with_indexed_obj(sample_obj):
    b = sample_obj.set_index(sample_obj.group + [sample_obj.timestamp_col])

def test_set_index(sample_obj):
    # print(isinstance(a, pd.DataFrame))
    assert sample_obj.set_index(sample_obj.group + [sample_obj.timestamp_col]).index.names == ['group', 'timestamp']

Comments

यदि संपत्ति के Noneलिए एक अमान्य मूल्य है group, तो क्या आपको नहीं उठाना चाहिए ValueError?

— चेपनर

आप सही हैं कि Noneएक अमान्य मूल्य है, यही कारण है कि मुझे यह कथन पसंद नहीं है। लेकिन यह जोड़ना कि कोई भी इसे परीक्षणों में शामिल नहीं करता है। मैं देख रहा हूँ कि कैसे अगर कोई भी बयान के बिना इसे ठीक से ठीक करने के लिए।

— पेज

सेटर को उठाना चाहिए ValueError। समस्या यह पता लगाना है कि पहली जगह में groupविशेषता को सेट करने के लिए क्या प्रयास किया जा रहा Noneहै।

— चेपनर

@chepner हाँ, बिल्कुल।

— पेज

शायद पंडों फ्लेवर पैकेज मदद कर सकता है।

— मायकोला जोतको

Answer 1

set_index()विधि कॉल करेंगे self.copy()अपने DataFrame वस्तु की एक प्रतिलिपि बनाने (स्रोत कोड को देखने के लिए आंतरिक रूप से यहाँ ) है, जो अंदर यह अपने अनुकूलित निर्माता विधि का उपयोग करता, _internal_ctor(), नई वस्तु (बनाने के लिए स्रोत )। ध्यान दें कि self._constructor()यह समान है self._internal_ctor(), जो गहरी-कॉपी या स्लाइसिंग जैसे संचालन के दौरान नए उदाहरण बनाने के लिए लगभग सभी पांडा कक्षाओं के लिए एक सामान्य आंतरिक विधि है। आपकी समस्या वास्तव में इस फ़ंक्शन से उत्पन्न होती है:

class HistDollarGains(pd.DataFrame):
    ...
    @classmethod
    def _internal_ctor(cls, *args, **kwargs):
        kwargs["group"]         = None
        kwargs["timestamp_col"] = None
        return cls(*args, **kwargs) # this is equivalent to calling
                                    # HistDollarGains(data, group=None, timestamp_col=None)

मुझे लगता है कि आपने इस कोड को github मुद्दे से कॉपी किया है । लाइनें kwargs["**"] = Noneस्पष्ट रूप से कंस्ट्रक्टर Noneको दोनों को सेट करने के लिए कहती हैं groupऔर timestamp_col। अंत में सेटर / सत्यापनकर्ता Noneनए मूल्य के रूप में मिलता है और एक त्रुटि उठाता है।

इसलिए, आपको एक स्वीकार्य मूल्य निर्धारित करना चाहिए groupऔर timestamp_col।

    @classmethod
    def _internal_ctor(cls, *args, **kwargs):
        kwargs["group"]         = []
        kwargs["timestamp_col"] = 'timestamp' # or whatever name that makes your validator happy
        return cls(*args, **kwargs)

फिर आप if g == None: returnसत्यापनकर्ता में लाइनों को हटा सकते हैं ।