找出数组中有多少组和小于k的三个数python版

给出一个长度为n的数组Arr,和一个数k,找出数组中有多少组和小于k的三个数，即Arr[a]+Arr[b]+arr[c]<k且0<=a<b<c<n的三元组的个数
输入：
6（第一行为一个整数n，代表输入数组的长度）
-2 0 1 2 3 6(第二行为n个整数)
2（第三行为一个整数k）
输出：
4（一个整数，表示有多少组满足条件的三元组）

import sys
num = int(sys.stdin.readline().strip().split()[0])
arr = map(int, sys.stdin.readline().strip().split())
k = int(sys.stdin.readline().strip().split()[0])
sorted(arr)
print arr
count = 0
for index in range(num):
    i = index + 1
    j = num-1
    while i < j:
        if (arr[i] + arr[j] < k - arr[index]):
            # print i,j,index
            count+=1
            print i,j,index
            j-=1
        else:
            j -= 1
        if i==j:
            i+=1
            j=num-1
print count