Skip to content

Commit 338ab2f

Browse files
hoostuspolarmutex
authored andcommitted
remove pandas chained assignments #91
1 parent 66806e4 commit 338ab2f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/fava_envelope/modules/beancount_envelope.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -153,29 +153,29 @@ def envelope_tables(self):
153153
if any(regexp.match(row[0]) for regexp in self.budget_accounts):
154154
if len(row) > 1 and row[1] is not None:
155155
starting_balance += row[1]
156-
self.income_df[months[0]]["Avail Income"] += starting_balance
156+
self.income_df.loc["Avail Income", months[0]] += starting_balance
157157

158158
self.envelope_df.fillna(Decimal(0.00), inplace=True)
159159

160160
# Set available
161161
for index, row in self.envelope_df.iterrows():
162162
for index2, month in enumerate(months):
163163
if index2 == 0:
164-
self.envelope_df[month, "available"][index] = (
164+
self.envelope_df.loc[index, (month, "available")] = (
165165
row[month, "budgeted"] + row[month, "activity"]
166166
)
167167
else:
168168
prev_available = self.envelope_df[
169169
months[index2 - 1], "available"
170170
][index]
171171
if prev_available > 0 or self.negative_rollover:
172-
self.envelope_df[month, "available"][index] = (
172+
self.envelope_df.loc[index, (month, "available")] = (
173173
prev_available
174174
+ row[month, "budgeted"]
175175
+ row[month, "activity"]
176176
)
177177
else:
178-
self.envelope_df[month, "available"][index] = (
178+
self.envelope_df.loc[index, (month, "available")] = (
179179
row[month, "budgeted"] + row[month, "activity"]
180180
)
181181

0 commit comments

Comments
 (0)