compile R online

Language: Layout:

goodExample <- T

#обозначение центроида
joke <- 17

x <- c(2.1, 5.5, 6.0, 6.5, 8.1, 9.0, 9.5, 4.0, 5.0, 7.6, 6.7, 3.1, 2.3, 7.0, 5.0, 6.5)
y <- c(8.0, 7.6, 3.3, 2.7, 3.1, 6.7, 4.6, 6.5, 3.5, 2.9, 8.0, 5.7, 5.5, 8.1, 8.7, 7.6)

df <- data.frame(x, y)

if (goodExample) {
    #хорошие начальные центры
    pGreen <- list(x = 3.1, y = 5.7)
    pBlue  <- list(x = 6.5, y = 2.7)
    pRed   <- list(x = 9.5, y = 4.6)
} else {
    #плохие начальные центры
    pGreen <- list(x = 4.0, y = 6.5)
    pBlue  <- list(x = 6.0, y = 3.3)
    pRed   <- list(x = 6.5, y = 7.6)
}

df$color <- 'black'

plot(
    df$x, df$y, 
    type = 'p', pch = 20, col = df$color, 
    xlim = c(min(df$x) * 0.80, max(df$x) * 1.20), ylim = c(min(df$y) * 0.80, max(df$y) * 1.20), 
    xaxt = 'n', yaxt = 'n', 
    ann = FALSE, cex = 1.5
)
lines(pGreen$x, pGreen$y, type = 'p', pch = joke, col = 'green', cex = 1.5)
lines(pBlue$x,  pBlue$y,  type = 'p', pch = joke, col = 'blue',  cex = 1.5)
lines(pRed$x,   pRed$y,   type = 'p', pch = joke, col = 'Red',   cex = 1.5)

for (i in 1:nrow(df)){
    if (all(df[i, c('x', 'y')] == pGreen)) {
        df[i, 'color'] <- 'green'
    } else if (all(df[i, c('x', 'y')] == pBlue)) {
        df[i, 'color'] <- 'blue'
    } else if (all(df[i, c('x', 'y')] == pRed)) {
        df[i, 'color'] <- 'red'
    }
}
       
        
df$toGreen <- 0
df$toBlue  <- 0
df$toRed   <- 0
        
norma <- function(p1, p2){
    res <- (p1$x - p2$x) ** 2 + (p1$y - p2$y) ** 2
    res <- sqrt(res)
    return(res)
}

df$toGreen <- norma(df[, c('x', 'y')], pGreen)
df$toBlue  <- norma(df[, c('x', 'y')], pBlue)
df$toRed   <- norma(df[, c('x', 'y')], pRed)
        
df$minDist <- apply(df[, c('toGreen', 'toBlue', 'toRed')], 1, min)

df$color <- ifelse(
    df$minDist == df$toGreen, 
    'green', 
    ifelse(
        df$minDist == df$toBlue, 
        'blue', 
        ifelse(
            df$minDist == df$toRed, 
            'red', 
            'black'
        )
    )
)
        
plot(
    x, y, 
    type = 'p', pch = 20, col = df$color, 
    xlim = c(min(df$x) * 0.80, max(df$x) * 1.20), ylim = c(min(df$y) * 0.80, max(df$y) * 1.20), 
    xaxt = 'n', yaxt = 'n', 
    ann = FALSE, cex = 1.5
)
lines(pGreen$x, pGreen$y, type = 'p', pch = joke, col = 'green', cex = 1.5)
lines(pBlue$x,  pBlue$y,  type = 'p', pch = joke, col = 'blue',  cex = 1.5)
lines(pRed$x,   pRed$y,   type = 'p', pch = joke, col = 'Red',   cex = 1.5)

pGreen <- list(x = mean(subset(df$x, df$color == 'green')), y = mean(subset(df$y, df$color == 'green')))
pBlue  <- list(x = mean(subset(df$x, df$color == 'blue')),  y = mean(subset(df$y, df$color == 'blue')))
pRed   <- list(x = mean(subset(df$x, df$color == 'red')),   y = mean(subset(df$y, df$color == 'red')))
        
plot(
    df$x, df$y, 
    type = 'p', pch = 20, col = df$color, 
    xlim = c(min(df$x) * 0.80, max(df$x) * 1.20), ylim = c(min(df$y) * 0.80, max(df$y) * 1.20), 
    xaxt = 'n', yaxt = 'n', 
    ann = FALSE, cex = 1.5
)
lines(pGreen$x, pGreen$y, type = 'p', pch = joke, col = 'green', cex = 1.5)
lines(pBlue$x,  pBlue$y,  type = 'p', pch = joke, col = 'blue',  cex = 1.5)
lines(pRed$x,   pRed$y,   type = 'p', pch = joke, col = 'Red',   cex = 1.5)
        
pGreenPrev <- pGreen
pBluePrev  <- pBlue
pRedPrev   <- pRed

i = 1
        
repeat{
    
    #break
    
    i = i + 1
       
    df$toGreen <- norma(df[, c('x', 'y')], pGreen)
    df$toBlue  <- norma(df[, c('x', 'y')], pBlue)
    df$toRed   <- norma(df[, c('x', 'y')], pRed)
    
    df$minDist <- apply(df[, c('toGreen', 'toBlue', 'toRed')], 1, min)
    
    df$color <- ifelse(
        df$minDist == df$toGreen, 
        'green', 
        ifelse(
            df$minDist == df$toBlue, 
            'blue', 
            ifelse(
                df$minDist == df$toRed, 
                'red', 
                'black'
            )
        )
    )
    
    pGreen <- list(x = mean(subset(df$x, df$color == 'green')), y = mean(subset(df$y, df$color == 'green')))
    pBlue  <- list(x = mean(subset(df$x, df$color == 'blue')),  y = mean(subset(df$y, df$color == 'blue')))
    pRed   <- list(x = mean(subset(df$x, df$color == 'red')),   y = mean(subset(df$y, df$color == 'red')))
    
    if (identical(pGreenPrev, pGreen) && identical(pBluePrev, pBlue) && identical(pRedPrev, pRed)) break
    
    plot(
        df$x, df$y, 
        type = 'p', pch = 20, col = df$color, 
        xlim = c(min(df$x) * 0.80, max(df$x) * 1.20), ylim = c(min(df$y) * 0.80, max(df$y) * 1.20), 
        xaxt = 'n', yaxt = 'n', 
        ann = FALSE, cex = 1.5
    )
    lines(pGreen$x, pGreen$y, type = 'p', pch = joke, col = 'green', cex = 1.5)
    lines(pBlue$x,  pBlue$y,  type = 'p', pch = joke, col = 'blue',  cex = 1.5)
    lines(pRed$x,   pRed$y,   type = 'p', pch = joke, col = 'Red',   cex = 1.5)
    
    #if (i > 5) break
        
    pGreenPrev <- pGreen
    pBluePrev  <- pBlue
    pRedPrev   <- pRed
}

[ + ] Show input
Absolute running time: 0.8 sec, cpu time: 0.36 sec, memory peak: 36 Mb, absolute service time: 0.84 sec