@@ -109,7 +109,7 @@ class _Data:
109
109
"""
110
110
def __init__ (self , df : pd .DataFrame ):
111
111
self .__df = df
112
- self .__i = len (df )
112
+ self .__len = len (df ) # Current length
113
113
self .__pip : Optional [float ] = None
114
114
self .__cache : Dict [str , _Array ] = {}
115
115
self .__arrays : Dict [str , _Array ] = {}
@@ -124,8 +124,8 @@ def __getattr__(self, item):
124
124
except KeyError :
125
125
raise AttributeError (f"Column '{ item } ' not in data" ) from None
126
126
127
- def _set_length (self , i ):
128
- self .__i = i
127
+ def _set_length (self , length ):
128
+ self .__len = length
129
129
self .__cache .clear ()
130
130
131
131
def _update (self ):
@@ -136,18 +136,18 @@ def _update(self):
136
136
self .__arrays ['__index' ] = index
137
137
138
138
def __repr__ (self ):
139
- i = min (self .__i , len (self .__df )) - 1
139
+ i = min (self .__len , len (self .__df )) - 1
140
140
index = self .__arrays ['__index' ][i ]
141
141
items = ', ' .join (f'{ k } ={ v } ' for k , v in self .__df .iloc [i ].items ())
142
142
return f'<Data i={ i } ({ index } ) { items } >'
143
143
144
144
def __len__ (self ):
145
- return self .__i
145
+ return self .__len
146
146
147
147
@property
148
148
def df (self ) -> pd .DataFrame :
149
- return (self .__df .iloc [:self .__i ]
150
- if self .__i < len (self .__df )
149
+ return (self .__df .iloc [:self .__len ]
150
+ if self .__len < len (self .__df )
151
151
else self .__df )
152
152
153
153
@property
@@ -160,7 +160,7 @@ def pip(self) -> float:
160
160
def __get_array (self , key ) -> _Array :
161
161
arr = self .__cache .get (key )
162
162
if arr is None :
163
- arr = self .__cache [key ] = cast (_Array , self .__arrays [key ][:self .__i ])
163
+ arr = self .__cache [key ] = cast (_Array , self .__arrays [key ][:self .__len ])
164
164
return arr
165
165
166
166
@property
0 commit comments