|
26 | 26 | }, |
27 | 27 | { |
28 | 28 | "cell_type": "code", |
29 | | - "execution_count": 18, |
| 29 | + "execution_count": 6, |
30 | 30 | "id": "9a63283cbaf04dbcab1f6479b197f3a8", |
31 | 31 | "metadata": { |
32 | 32 | "editable": true |
|
48 | 48 | }, |
49 | 49 | { |
50 | 50 | "cell_type": "code", |
51 | | - "execution_count": null, |
| 51 | + "execution_count": 7, |
52 | 52 | "id": "72eea5119410473aa328ad9291626812", |
53 | 53 | "metadata": { |
54 | 54 | "editable": true |
|
63 | 63 | " │ --- ┆ --- ┆ --- ┆ --- ┆ --- │\n", |
64 | 64 | " │ i64 ┆ str ┆ i64 ┆ f64 ┆ i64 │\n", |
65 | 65 | " ╞══════╪═════════════════════════════════╪═══════╪══════════════╪══════════╡\n", |
66 | | - " │ 2 ┆ 162681765859364298619846106603… ┆ 0 ┆ 1000.0 ┆ 1000 │\n", |
67 | | - " │ 4 ┆ 162681765859364298619846106603… ┆ 0 ┆ 1000.0 ┆ 1000 │\n", |
68 | | - " │ 6 ┆ 162681765859364298619846106603… ┆ 0 ┆ 1000.0 ┆ 1000 │\n", |
69 | | - " │ 8 ┆ 162681765859364298619846106603… ┆ 0 ┆ 1000.0 ┆ 1000 │\n", |
70 | | - " │ 10 ┆ 162681765859364298619846106603… ┆ 0 ┆ 1000.0 ┆ 1000 │\n", |
| 66 | + " │ 2 ┆ 332212815818606584686857770936… ┆ 0 ┆ 1000.0 ┆ 1000 │\n", |
| 67 | + " │ 4 ┆ 332212815818606584686857770936… ┆ 0 ┆ 1000.0 ┆ 1000 │\n", |
| 68 | + " │ 6 ┆ 332212815818606584686857770936… ┆ 0 ┆ 1000.0 ┆ 1000 │\n", |
| 69 | + " │ 8 ┆ 332212815818606584686857770936… ┆ 0 ┆ 1000.0 ┆ 1000 │\n", |
| 70 | + " │ 10 ┆ 332212815818606584686857770936… ┆ 0 ┆ 1000.0 ┆ 1000 │\n", |
71 | 71 | " └──────┴─────────────────────────────────┴───────┴──────────────┴──────────┘,\n", |
72 | 72 | " 'agent': shape: (5_000, 4)\n", |
73 | 73 | " ┌────────────────────┬──────┬─────────────────────────────────┬───────┐\n", |
74 | 74 | " │ wealth_MoneyAgents ┆ step ┆ seed ┆ batch │\n", |
75 | 75 | " │ --- ┆ --- ┆ --- ┆ --- │\n", |
76 | 76 | " │ f64 ┆ i32 ┆ str ┆ i32 │\n", |
77 | 77 | " ╞════════════════════╪══════╪═════════════════════════════════╪═══════╡\n", |
78 | | - " │ 0.0 ┆ 2 ┆ 162681765859364298619846106603… ┆ 0 │\n", |
79 | | - " │ 3.0 ┆ 2 ┆ 162681765859364298619846106603… ┆ 0 │\n", |
80 | | - " │ 1.0 ┆ 2 ┆ 162681765859364298619846106603… ┆ 0 │\n", |
81 | | - " │ 3.0 ┆ 2 ┆ 162681765859364298619846106603… ┆ 0 │\n", |
82 | | - " │ 6.0 ┆ 2 ┆ 162681765859364298619846106603… ┆ 0 │\n", |
| 78 | + " │ 3.0 ┆ 2 ┆ 332212815818606584686857770936… ┆ 0 │\n", |
| 79 | + " │ 0.0 ┆ 2 ┆ 332212815818606584686857770936… ┆ 0 │\n", |
| 80 | + " │ 2.0 ┆ 2 ┆ 332212815818606584686857770936… ┆ 0 │\n", |
| 81 | + " │ 1.0 ┆ 2 ┆ 332212815818606584686857770936… ┆ 0 │\n", |
| 82 | + " │ 0.0 ┆ 2 ┆ 332212815818606584686857770936… ┆ 0 │\n", |
83 | 83 | " │ … ┆ … ┆ … ┆ … │\n", |
84 | | - " │ 4.0 ┆ 10 ┆ 162681765859364298619846106603… ┆ 0 │\n", |
85 | | - " │ 1.0 ┆ 10 ┆ 162681765859364298619846106603… ┆ 0 │\n", |
86 | | - " │ 0.0 ┆ 10 ┆ 162681765859364298619846106603… ┆ 0 │\n", |
87 | | - " │ 0.0 ┆ 10 ┆ 162681765859364298619846106603… ┆ 0 │\n", |
88 | | - " │ 0.0 ┆ 10 ┆ 162681765859364298619846106603… ┆ 0 │\n", |
| 84 | + " │ 0.0 ┆ 10 ┆ 332212815818606584686857770936… ┆ 0 │\n", |
| 85 | + " │ 0.0 ┆ 10 ┆ 332212815818606584686857770936… ┆ 0 │\n", |
| 86 | + " │ 0.0 ┆ 10 ┆ 332212815818606584686857770936… ┆ 0 │\n", |
| 87 | + " │ 0.0 ┆ 10 ┆ 332212815818606584686857770936… ┆ 0 │\n", |
| 88 | + " │ 0.0 ┆ 10 ┆ 332212815818606584686857770936… ┆ 0 │\n", |
89 | 89 | " └────────────────────┴──────┴─────────────────────────────────┴───────┘}" |
90 | 90 | ] |
91 | 91 | }, |
92 | | - "execution_count": 19, |
| 92 | + "execution_count": 7, |
93 | 93 | "metadata": {}, |
94 | 94 | "output_type": "execute_result" |
95 | 95 | } |
|
175 | 175 | }, |
176 | 176 | { |
177 | 177 | "cell_type": "code", |
178 | | - "execution_count": null, |
| 178 | + "execution_count": 8, |
179 | 179 | "id": "5f14f38c", |
180 | 180 | "metadata": {}, |
181 | 181 | "outputs": [ |
|
185 | 185 | "[]" |
186 | 186 | ] |
187 | 187 | }, |
188 | | - "execution_count": 20, |
| 188 | + "execution_count": 8, |
189 | 189 | "metadata": {}, |
190 | 190 | "output_type": "execute_result" |
191 | 191 | } |
|
228 | 228 | }, |
229 | 229 | { |
230 | 230 | "cell_type": "code", |
231 | | - "execution_count": null, |
| 231 | + "execution_count": 9, |
232 | 232 | "id": "8763a12b2bbd4a93a75aff182afb95dc", |
233 | 233 | "metadata": { |
234 | 234 | "editable": true |
|
240 | 240 | "[]" |
241 | 241 | ] |
242 | 242 | }, |
243 | | - "execution_count": 21, |
| 243 | + "execution_count": 9, |
244 | 244 | "metadata": {}, |
245 | 245 | "output_type": "execute_result" |
246 | 246 | } |
|
283 | 283 | }, |
284 | 284 | { |
285 | 285 | "cell_type": "code", |
286 | | - "execution_count": null, |
| 286 | + "execution_count": 10, |
287 | 287 | "id": "7cdc8c89c7104fffa095e18ddfef8986", |
288 | 288 | "metadata": { |
289 | 289 | "editable": true |
|
293 | 293 | "model_s3 = MoneyModel(1000)\n", |
294 | 294 | "model_s3.dc = DataCollector(\n", |
295 | 295 | " model=model_s3,\n", |
296 | | - " model_reporters = {\n", |
297 | | - "\"total_wealth\": lambda m: sum(\n", |
298 | | - "s[\"wealth\"].sum() for s in m.sets if \"wealth\" in s.df.columns\n", |
299 | | - "),\n", |
300 | | - "\"n_agents\": lambda m: len(m.sets.ids),\n", |
301 | | - "}\n", |
302 | | - "\n", |
303 | | - "\n", |
304 | | - " \"total_wealth\": lambda m: list(m.sets.df.values())[0][\"wealth\"].sum(),\n", |
305 | | - " \"n_agents\": lambda m: len(list(m.sets.df.values())[0]),\n", |
| 296 | + " model_reporters={\n", |
| 297 | + " \"total_wealth\": lambda m: sum(\n", |
| 298 | + " s[\"wealth\"].sum() for s in m.sets if \"wealth\" in s.df.columns\n", |
| 299 | + " ),\n", |
| 300 | + " \"n_agents\": lambda m: len(m.sets.ids),\n", |
306 | 301 | " },\n", |
307 | 302 | " agent_reporters={\n", |
308 | 303 | " \"wealth\": \"wealth\",\n", |
|
330 | 325 | }, |
331 | 326 | { |
332 | 327 | "cell_type": "code", |
333 | | - "execution_count": 23, |
| 328 | + "execution_count": 11, |
334 | 329 | "id": "938c804e27f84196a10c8828c723f798", |
335 | 330 | "metadata": { |
336 | 331 | "editable": true |
|
392 | 387 | }, |
393 | 388 | { |
394 | 389 | "cell_type": "code", |
395 | | - "execution_count": 24, |
| 390 | + "execution_count": 12, |
396 | 391 | "id": "59bbdb311c014d738909a11f9e486628", |
397 | 392 | "metadata": { |
398 | 393 | "editable": true |
|
421 | 416 | }, |
422 | 417 | { |
423 | 418 | "cell_type": "code", |
424 | | - "execution_count": 25, |
| 419 | + "execution_count": 13, |
425 | 420 | "id": "8a65eabff63a45729fe45fb5ade58bdc", |
426 | 421 | "metadata": { |
427 | 422 | "editable": true |
|
437 | 432 | " white-space: pre-wrap;\n", |
438 | 433 | "}\n", |
439 | 434 | "</style>\n", |
440 | | - "<small>shape: (5, 5)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>step</th><th>seed</th><th>batch</th><th>total_wealth</th><th>n_agents</th></tr><tr><td>i64</td><td>str</td><td>i64</td><td>f64</td><td>i64</td></tr></thead><tbody><tr><td>2</td><td>"732054881101029867447298951813…</td><td>0</td><td>100.0</td><td>100</td></tr><tr><td>4</td><td>"732054881101029867447298951813…</td><td>0</td><td>100.0</td><td>100</td></tr><tr><td>6</td><td>"732054881101029867447298951813…</td><td>0</td><td>100.0</td><td>100</td></tr><tr><td>8</td><td>"732054881101029867447298951813…</td><td>0</td><td>100.0</td><td>100</td></tr><tr><td>10</td><td>"732054881101029867447298951813…</td><td>0</td><td>100.0</td><td>100</td></tr></tbody></table></div>" |
| 435 | + "<small>shape: (5, 5)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>step</th><th>seed</th><th>batch</th><th>total_wealth</th><th>n_agents</th></tr><tr><td>i64</td><td>str</td><td>i64</td><td>f64</td><td>i64</td></tr></thead><tbody><tr><td>2</td><td>"540832786058427425452319829502…</td><td>0</td><td>100.0</td><td>100</td></tr><tr><td>4</td><td>"540832786058427425452319829502…</td><td>0</td><td>100.0</td><td>100</td></tr><tr><td>6</td><td>"540832786058427425452319829502…</td><td>0</td><td>100.0</td><td>100</td></tr><tr><td>8</td><td>"540832786058427425452319829502…</td><td>0</td><td>100.0</td><td>100</td></tr><tr><td>10</td><td>"540832786058427425452319829502…</td><td>0</td><td>100.0</td><td>100</td></tr></tbody></table></div>" |
441 | 436 | ], |
442 | 437 | "text/plain": [ |
443 | 438 | "shape: (5, 5)\n", |
|
446 | 441 | "│ --- ┆ --- ┆ --- ┆ --- ┆ --- │\n", |
447 | 442 | "│ i64 ┆ str ┆ i64 ┆ f64 ┆ i64 │\n", |
448 | 443 | "╞══════╪═════════════════════════════════╪═══════╪══════════════╪══════════╡\n", |
449 | | - "│ 2 ┆ 732054881101029867447298951813… ┆ 0 ┆ 100.0 ┆ 100 │\n", |
450 | | - "│ 4 ┆ 732054881101029867447298951813… ┆ 0 ┆ 100.0 ┆ 100 │\n", |
451 | | - "│ 6 ┆ 732054881101029867447298951813… ┆ 0 ┆ 100.0 ┆ 100 │\n", |
452 | | - "│ 8 ┆ 732054881101029867447298951813… ┆ 0 ┆ 100.0 ┆ 100 │\n", |
453 | | - "│ 10 ┆ 732054881101029867447298951813… ┆ 0 ┆ 100.0 ┆ 100 │\n", |
| 444 | + "│ 2 ┆ 540832786058427425452319829502… ┆ 0 ┆ 100.0 ┆ 100 │\n", |
| 445 | + "│ 4 ┆ 540832786058427425452319829502… ┆ 0 ┆ 100.0 ┆ 100 │\n", |
| 446 | + "│ 6 ┆ 540832786058427425452319829502… ┆ 0 ┆ 100.0 ┆ 100 │\n", |
| 447 | + "│ 8 ┆ 540832786058427425452319829502… ┆ 0 ┆ 100.0 ┆ 100 │\n", |
| 448 | + "│ 10 ┆ 540832786058427425452319829502… ┆ 0 ┆ 100.0 ┆ 100 │\n", |
454 | 449 | "└──────┴─────────────────────────────────┴───────┴──────────────┴──────────┘" |
455 | 450 | ] |
456 | 451 | }, |
457 | | - "execution_count": 25, |
| 452 | + "execution_count": 13, |
458 | 453 | "metadata": {}, |
459 | 454 | "output_type": "execute_result" |
460 | 455 | } |
|
0 commit comments