|
332 | 332 | "role_dept_list = []\n",
|
333 | 333 | "role_field_list = []\n",
|
334 | 334 | "\n",
|
| 335 | + "edge_emp_dept_rows_list = [edge_emp_dept]\n", |
| 336 | + "edge_emp_role_rows_list = [edge_emp_role]\n", |
| 337 | + "edge_emp_field_rows_list = [edge_emp_field]\n", |
| 338 | + "edge_role_dept_rows_list = [edge_role_dept]\n", |
| 339 | + "edge_role_field_rows_list = [edge_role_field]\n", |
| 340 | + "\n", |
335 | 341 | "for index, row in df.iterrows():\n",
|
336 | 342 | " emp = row['EmployeeNumber']\n",
|
337 | 343 | " emp_id = emp_map[emp]\n",
|
|
341 | 347 | " field_id = field_map[field]\n",
|
342 | 348 | " dept = row['Department']\n",
|
343 | 349 | " dept_id = dept_map[dept]\n",
|
344 |
| - " \n", |
345 |
| - " edge_emp_dept = edge_emp_dept.append({'~id': uuid.uuid4(), '~from': emp_id, \n", |
346 |
| - " '~to': dept_id, \n", |
347 |
| - " '~label': 'works_in'}, ignore_index=True)\n", |
348 |
| - " edge_emp_role = edge_emp_role.append({'~id': uuid.uuid4(), '~from': emp_id, \n", |
349 |
| - " '~to': role_id, \n", |
350 |
| - " '~label': 'works_as'}, ignore_index=True)\n", |
351 |
| - " edge_emp_field = edge_emp_field.append({'~id': uuid.uuid4(), '~from': emp_id, \n", |
352 |
| - " '~to': field_id, \n", |
353 |
| - " '~label': 'has_education_level'}, ignore_index=True)\n", |
| 350 | + "\n", |
| 351 | + " edge_emp_dept_row_df = pd.DataFrame.from_dict({'~id': uuid.uuid4(),\n", |
| 352 | + " '~from': emp_id,\n", |
| 353 | + " '~to': dept_id,\n", |
| 354 | + " '~label': 'works_in'},\n", |
| 355 | + " orient='index').T\n", |
| 356 | + " edge_emp_dept_rows_list.append(edge_emp_dept_row_df)\n", |
| 357 | + " edge_emp_role_row_df = pd.DataFrame.from_dict({'~id': uuid.uuid4(),\n", |
| 358 | + " '~from': emp_id,\n", |
| 359 | + " '~to': role_id,\n", |
| 360 | + " '~label': 'works_as'},\n", |
| 361 | + " orient='index').T\n", |
| 362 | + " edge_emp_role_rows_list.append(edge_emp_role_row_df)\n", |
| 363 | + " edge_emp_field_row_df = pd.DataFrame.from_dict({'~id': uuid.uuid4(),\n", |
| 364 | + " '~from': emp_id,\n", |
| 365 | + " '~to': field_id,\n", |
| 366 | + " '~label': 'has_education_level'},\n", |
| 367 | + " orient='index').T\n", |
| 368 | + " edge_emp_field_rows_list.append(edge_emp_field_row_df)\n", |
354 | 369 | " \n",
|
355 | 370 | " role_dept = f\"{role_id}-{dept_id}\"\n",
|
356 | 371 | " role_field = f\"{role_id}-{field_id}\"\n",
|
357 | 372 | " if role_dept not in role_dept_list:\n",
|
358 |
| - " edge_role_dept = edge_role_dept.append({'~id': uuid.uuid4(), '~from': role_id, \n", |
359 |
| - " '~to': dept_id, \n", |
360 |
| - " '~label': 'part_of'}, ignore_index=True)\n", |
| 373 | + " edge_role_dept_row_df = pd.DataFrame.from_dict({'~id': uuid.uuid4(),\n", |
| 374 | + " '~from': role_id,\n", |
| 375 | + " '~to': dept_id,\n", |
| 376 | + " '~label': 'part_of'},\n", |
| 377 | + " orient='index').T\n", |
| 378 | + " edge_role_dept_rows_list.append(edge_role_dept_row_df)\n", |
| 379 | + " #edge_role_dept = pd.concat([edge_role_dept, edge_role_dept_row_df], ignore_index=True)\n", |
361 | 380 | " role_dept_list.append(role_dept)\n",
|
362 | 381 | " if role_field not in role_field_list:\n",
|
363 |
| - " edge_role_field = edge_role_field.append({'~id': uuid.uuid4(), '~from': role_id, \n", |
364 |
| - " '~to': field_id, \n", |
365 |
| - " '~label': 'requires'}, ignore_index=True)\n", |
| 382 | + " edge_role_field_row_df = pd.DataFrame.from_dict({'~id': uuid.uuid4(), '~from': role_id,\n", |
| 383 | + " '~to': field_id,\n", |
| 384 | + " '~label': 'requires'},\n", |
| 385 | + " orient='index').T\n", |
| 386 | + " edge_role_field_rows_list.append(edge_role_field_row_df)\n", |
366 | 387 | " role_field_list.append(role_field)\n",
|
367 | 388 | " edge_cnt = edge_cnt + 1\n",
|
| 389 | + "\n", |
| 390 | + "edge_emp_dept = pd.concat(edge_emp_dept_rows_list, ignore_index=True)\n", |
| 391 | + "edge_emp_role = pd.concat(edge_emp_role_rows_list, ignore_index=True)\n", |
| 392 | + "edge_emp_field = pd.concat(edge_emp_field_rows_list, ignore_index=True)\n", |
| 393 | + "edge_role_dept = pd.concat(edge_role_dept_rows_list, ignore_index=True)\n", |
| 394 | + "edge_role_field = pd.concat(edge_role_field_rows_list, ignore_index=True)\n", |
| 395 | + "\n", |
368 | 396 | "edge_df = pd.concat([edge_emp_dept, edge_emp_role, edge_emp_field, edge_role_dept, edge_role_field])\n",
|
369 | 397 | "edge_df.to_csv(os.path.join(output_folder, 'edge.csv'), index=False)\n",
|
370 | 398 | "\n",
|
|
0 commit comments