Pandas 와 Numpy 를 활용하여 Rating Matrix 와 Frequent Matrix 를 만드는 과제 <- 파일 다운로드 가능!

---

🌼 정답 코드

[Data Handling] Assignment :: Lab Build Matrix

https://github.com/TEAMLAB-Lecture/AI-python-connect/blob/master/lab_assignments/lab_3/README.md

import numpy as np
import pandas as pd

def get_rating_matrix(filename, dtype=np.float32):
    df = pd.read_csv(filename)
    return df.groupby(["source", "target"])["rating"].sum().unstack().fillna(0)

get_rating_matrix("movie_rating.csv")

def get_frequent_matrix(filename, dtype=np.float32):
    df = pd.read_csv(filename)
    df["rating"]=1  #"rating"이 없기 때문에 모두 1로 이루어진 rating column을 생성해줌
    return df.groupby(["source", "target"])["rating"].sum().unstack().fillna(0)

get_frequent_matrix("1000i.csv")