인프런 커뮤니티 질문&답변

안녕하세요.

처음하는 머신러닝이라 문제를 풀어서 통과를 했는데, 맞게 한지 모르겠습니다.

아래처럼 하면 되는지 확인 부탁드립니다.

import numpy as np

import pandas as pd

from pandas import DataFrame

def get_rating_matrix(filename, dtype=np.float32):

df = pd.read_csv(filename)

user = sorted(df.source.unique())

movie = sorted(df.target.unique())

data = {mv: {us[0]: us[1] for us in df[["source","rating"]][df["target"] == mv].values} for mv in movie}

df_data = DataFrame(data, index=user).fillna(0)

result = df_data.values

return result

def get_frequent_matrix(filename, dtype=np.float32):

df = pd.read_csv(filename)

index_name = sorted(df.source.unique())

columns_name = sorted(df.target.unique())

data = { idx : { t[0]:t[1] for t in zip(df["target"][df["source"] == idx].value_counts().index,  df["target"][df["source"] == idx].value_counts().values) } for idx in index_name }
df2 = DataFrame(data, dtype=np.float32)
result = df2.T.values
return result