java 版本

package com.fullshare.test;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;

/**
 * @author huangjiangnan
 * @email [email protected]
 * @version 1.0
 * @since 2017年12月14日 上午10:27:35 類說明
 */
public class Knn {

	public static void main(String[] args) {
		// 電影名稱 搞笑鏡頭 擁抱鏡頭 打鬥鏡頭 電影型別
		Object[][] sample = { 
				{ 1, "寶貝當家", 45, 2, 9, "喜劇片" },
				{ 2, "美人魚", 21, 17, 5, "喜劇片" },
				{ 3, "澳門風雲3", 54, 9, 11, "喜劇片" }, 
				{ 4, "功夫熊貓3", 39, 0, 31, "喜劇片" }, 
				{ 5, "諜影重重", 5, 2, 57, "動作片" },
				{ 6, "葉問3", 3, 2, 65, "動作片" }, 
				{ 7, "倫敦陷落", 2, 3, 55, "動作片" }, 
				{ 8, "我的特工爺爺", 6, 4, 21, "動作片" }, 
				{ 9, "奔愛", 7, 46, 4, "愛情片" }, 
				{ 10, "夜孔雀", 9, 39, 8, "愛情片" },
				{ 11, "代理情人", 9, 38, 2, "愛情片" }, 
				{ 12, "新步步驚心", 8, 34, 17, "愛情片" }, };
		// 求唐人街辦案型別
		Object[] movie = { 13, "唐人街探案", 23, 3, 17, null };
		int length = sample.length;
		System.out.println("序號 名稱        距離");
		List<MovieDis> movieDisList = new ArrayList<>();

		for (int i = 0; i < length; i++) {
			Object[] mv = sample[i];
			double distances = getDistance(mv, movie);
			MovieDis movieDis = new MovieDis((int) mv[0], (String) mv[1], distances, (String) mv[5]);
			// Object[] disInfo={};
			System.out.println(String.format("%s %s %s", mv[0], mv[1], distances));
			movieDisList.add(movieDis);
		}

		Collections.sort(movieDisList, new Comparator<MovieDis>() {

			@Override
			public int compare(MovieDis o1, MovieDis o2) {
				double sub = (o1.getDistance() - o2.getDistance());
				if (sub == 0) {
					return 0;
				}
				if (sub > 0) {
					return 1;
				}
				return -1;
			}
		});
		
		int k=5;
		System.out.println("按照歐式距離排序，取k=5");
		movieDisList=movieDisList.subList(0,k);
		for (MovieDis movieDis : movieDisList) {
			System.out.println(movieDis);
		}

	}

	public static double getDistance(Object[] movie1, Object[] movie2) {
		double[] ps1 = { (Integer) movie1[2], (Integer) movie1[3], (Integer) movie1[4] };
		double[] ps2 = { (Integer) movie2[2], (Integer) movie2[3], (Integer) movie2[4] };
		return getDistance(ps1, ps2);
	}

	public static double getDistance(double[] ps1, double[] ps2) {
		if (ps1.length != ps1.length) {
			throw new RuntimeException("屬性數量不對應");
		}
		int length = ps1.length;
		double total = 0;
		for (int i = 0; i < length; i++) {
			double sub = ps1[i] - ps2[i];
			total = total + (sub * sub);
		}
		return Math.sqrt(total);
	}

}

class MovieDis {
	private int id;
	private String title;
	private double distance;
	private String type;

	public int getId() {
		return id;
	}

	public void setId(int id) {
		this.id = id;
	}

	public String getTitle() {
		return title;
	}

	public void setTitle(String title) {
		this.title = title;
	}

	public double getDistance() {
		return distance;
	}

	public void setDistance(double distance) {
		this.distance = distance;
	}

	public MovieDis(int id, String title, double distance, String type) {
		super();
		this.id = id;
		this.title = title;
		this.distance = distance;
		this.type = type;
	}


	@Override
	public String toString() {
		return "MovieDis [id=" + id + ", title=" + title + ", distance=" + distance + ", type=" + type + "]";
	}

	public String getType() {
		return type;
	}

	public void setType(String type) {
		this.type = type;
	}

}
 

package com.test.api

class KnnScala {

}

object KnnScala {
  def main(args: Array[String]): Unit = {
    var sample: Array[Array[Any]] = Array(
      Array(1, "寶貝當家", 45, 2, 9, "喜劇片"),
      Array(2, "美人魚", 21, 17, 5, "喜劇片"),
      Array(3, "澳門風雲3", 54, 9, 11, "喜劇片"),
      Array(4, "功夫熊貓3", 39, 0, 31, "喜劇片"),
      Array(5, "諜影重重", 5, 2, 57, "動作片"),
      Array(6, "葉問3", 3, 2, 65, "動作片"),
      Array(7, "倫敦陷落", 2, 3, 55, "動作片"),
      Array(8, "我的特工爺爺", 6, 4, 21, "動作片"),
      Array(9, "奔愛", 7, 46, 4, "愛情片"),
      Array(10, "夜孔雀", 9, 39, 8, "愛情片"),
      Array(11, "代理情人", 9, 38, 2, "愛情片"),
      Array(12, "新步步驚心", 8, 34, 17, "愛情片"))
    // 求唐人街辦案型別
    var movie = Array(13, "唐人街探案", 23, 3, 17, null);
    var length = sample.length - 1;
    println("序號 名稱        距離");
    var movieDisList = List[MovieDis]();

    for (i <- 0 to length) {
      var mv: Array[Any] = sample(i);
      var distances: Double = getDistance(mv, movie);
      var movieDis = new MovieDis(mv(0).asInstanceOf[Int], mv(1).asInstanceOf[String], distances, mv(5).asInstanceOf[String]);
      println(printf("%s %s %s", mv(0), mv(1), distances));
      //列表新增跟java不一樣，坑
      movieDisList = (movieDisList.+:(movieDis))
    }
    movieDisList = movieDisList.sortWith((o1: MovieDis, o2: MovieDis) => (o1.distance < o2.distance));

    var k: Int = 5;
    println("按照歐式距離排序，取k=5");
    movieDisList = movieDisList take 5;
    movieDisList.foreach { o => println(o) }
  }

  def getDistance(movie1: Array[Any], movie2: Array[Any]): Double = {
    var ps1 = Array(movie1(2).asInstanceOf[Integer].doubleValue(), movie1(3).asInstanceOf[Integer].doubleValue(), movie1(4).asInstanceOf[Integer].doubleValue());
    var ps2 = Array(movie2(2).asInstanceOf[Integer].doubleValue(), movie2(3).asInstanceOf[Integer].doubleValue(), movie2(4).asInstanceOf[Integer].doubleValue());
    return getDistance(ps1, ps2);
  }

  def getDistance(ps1: Array[Double], ps2: Array[Double]): Double = {
    if (ps1.length != ps1.length) {
      throw new RuntimeException("屬性數量不對應");
    }
    var length = ps1.length - 1;
    var total: Double = 0;
    for (i <- 0 to length) {
      var sub = ps1(i) - ps2(i);
      total = total + (sub * sub);
    }
    
    return Math.sqrt(total);
  }

}

class MovieDis extends Serializable {
  var id: Int = 0;
  var title: String = null;
  var distance: Double = 0;
  var movieType: String = null;

  def this(_id: Int, _title: String, _distance: Double, _movieType: String) {
    this();
    this.id = _id;
    this.title = _title;
    this.distance = _distance;
    this.movieType = _movieType;
  }

  override def toString(): String = {
    return "MovieDis [id=" + id + ", title=" + title + ", distance=" + distance + ", movieType=" + movieType + "]";
  }

}

 

#!/usr/bin/env python3
# -*- coding: UTF-8 -*-
# 給多個變數賦值
import math

class MovieDis :
    # 序號
    id = 0
    # 電影標題
    title = ""
    # 差距
    distance = 0.0
    # 電影型別
    movieType = ""

    def __init__(self, id ,title,distance,movieType):
        self.id=id
        self.title=title
        self.distance=distance
        self.movieType=movieType


    def __str__(self):
        return "MovieDis [id=%s, title=%s, distance=%s, movieType=%s]" % (self.id,self.title,self.distance,self.movieType)


def get_distance(ps1,ps2):
    if len(ps1) != len(ps2):
        raise Exception("陣列長度不匹配")
    ll = len(ps1)
    total=0
    for i in range(0,ll):
        sub = ps1[i] - ps2[i]
        total = total + (sub * sub)
    return math.sqrt(total)




sample = ((1, "寶貝當家", 45, 2, 9, "喜劇片"),
      (2, "美人魚", 21, 17, 5, "喜劇片"),
      (3, "澳門風雲3", 54, 9, 11, "喜劇片"),
      (4, "功夫熊貓3", 39, 0, 31, "喜劇片"),
      (5, "諜影重重", 5, 2, 57, "動作片"),
      (6, "葉問3", 3, 2, 65, "動作片"),
      (7, "倫敦陷落", 2, 3, 55, "動作片"),
      (8, "我的特工爺爺", 6, 4, 21, "動作片"),
      (9, "奔愛", 7, 46, 4, "愛情片"),
      (10, "夜孔雀", 9, 39, 8, "愛情片"),
      (11, "代理情人", 9, 38, 2, "愛情片"),
      (12, "新步步驚心", 8, 34, 17, "愛情片"))

movie = (13, "唐人街探案", 23, 3, 17, "")


length = len(sample) - 1
print("序號 名稱        距離")
movieDisList = []

for i in range(0,length):
    mv= sample[i]
    start=mv[2:5]
    end= movie[2:5]
    distances = get_distance(start, end)
    print("%s %s %s" % (mv[0],mv[1],distances))
    movieDis = MovieDis(mv[0], mv[1], distances, mv[5])
    movieDisList.append(movieDis)
movieDisList.sort(key=lambda x:x.distance)
for i in range(0,5):
    print(movieDisList[i])